pixaris package

Subpackages

• pixaris.data_loaders package
  • Submodules
  • pixaris.data_loaders.base module
    • DatasetLoader
      • DatasetLoader.load_dataset()
  • pixaris.data_loaders.gcp module
    • GCPDatasetLoader
      • GCPDatasetLoader.load_dataset()
  • pixaris.data_loaders.local module
    • LocalDatasetLoader
      • LocalDatasetLoader.load_dataset()
  • Module contents
• pixaris.experiment_handlers package
  • Submodules
  • pixaris.experiment_handlers.base module
    • ExperimentHandler
      • ExperimentHandler.load_experiment_results_for_dataset()
      • ExperimentHandler.load_images_for_experiment()
      • ExperimentHandler.load_projects_and_datasets()
      • ExperimentHandler.store_results()
  • pixaris.experiment_handlers.gcp module
    • GCPExperimentHandler
      • GCPExperimentHandler.load_experiment_results_for_dataset()
      • GCPExperimentHandler.load_images_for_experiment()
      • GCPExperimentHandler.load_projects_and_datasets()
      • GCPExperimentHandler.store_results()
  • pixaris.experiment_handlers.local module
    • LocalExperimentHandler
      • LocalExperimentHandler.load_experiment_results_for_dataset()
      • LocalExperimentHandler.load_images_for_experiment()
      • LocalExperimentHandler.load_projects_and_datasets()
      • LocalExperimentHandler.store_results()
  • Module contents
• pixaris.feedback_handlers package
  • Submodules
  • pixaris.feedback_handlers.base module
    • FeedbackHandler
      • FeedbackHandler.create_feedback_iteration()
      • FeedbackHandler.feedback_df
      • FeedbackHandler.feedback_iteration_choices
      • FeedbackHandler.get_feedback_per_image()
      • FeedbackHandler.load_all_feedback_iteration_info_for_project()
      • FeedbackHandler.load_images_for_feedback_iteration()
      • FeedbackHandler.load_projects_list()
      • FeedbackHandler.write_single_feedback()
  • pixaris.feedback_handlers.gcp module
    • GCPFeedbackHandler
      • GCPFeedbackHandler.create_feedback_iteration()
      • GCPFeedbackHandler.load_images_for_feedback_iteration()
      • GCPFeedbackHandler.load_projects_list()
      • GCPFeedbackHandler.write_single_feedback()
  • pixaris.feedback_handlers.local module
    • LocalFeedbackHandler
      • LocalFeedbackHandler.create_feedback_iteration()
      • LocalFeedbackHandler.load_images_for_feedback_iteration()
      • LocalFeedbackHandler.load_projects_list()
      • LocalFeedbackHandler.write_single_feedback()
  • Module contents
• pixaris.generation package
  • Submodules
  • pixaris.generation.base module
    • ImageGenerator
      • ImageGenerator.generate_single_image()
      • ImageGenerator.validate_inputs_and_parameters()
  • pixaris.generation.comfyui module
    • ComfyGenerator
      • ComfyGenerator.generate_single_image()
      • ComfyGenerator.validate_inputs_and_parameters()
  • pixaris.generation.comfyui_cluster module
    • ComfyClusterGenerator
      • ComfyClusterGenerator.close()
      • ComfyClusterGenerator.generate_single_image()
      • ComfyClusterGenerator.start_background_task()
      • ComfyClusterGenerator.update_available_hosts()
      • ComfyClusterGenerator.validate_inputs_and_parameters()
  • pixaris.generation.flux module
    • FluxFillGenerator
      • FluxFillGenerator.generate_single_image()
      • FluxFillGenerator.validate_inputs_and_parameters()
  • pixaris.generation.gemini module
    • GeminiGenerator
      • GeminiGenerator.generate_single_image()
      • GeminiGenerator.validate_inputs_and_parameters()
  • pixaris.generation.imagen2 module
    • Imagen2Generator
      • Imagen2Generator.generate_single_image()
      • Imagen2Generator.validate_inputs_and_parameters()
  • pixaris.generation.utils module
    • encode_image_to_bytes()
    • extract_value_from_list_of_dicts()
  • Module contents
• pixaris.metrics package
  • Submodules
  • pixaris.metrics.base module
    • BaseMetric
      • BaseMetric.calculate()
  • pixaris.metrics.iou module
    • IoUMetric
      • IoUMetric.calculate()
  • pixaris.metrics.llm module
    • BaseLLMMetric
      • BaseLLMMetric.calculate()
    • ErrorLLMMetric
      • ErrorLLMMetric.calculate()
    • SimilarityLLMMetric
      • SimilarityLLMMetric.calculate()
    • StyleLLMMetric
      • StyleLLMMetric.calculate()
  • pixaris.metrics.luminescence module
    • LuminescenceComparisonByMaskMetric
      • LuminescenceComparisonByMaskMetric.calculate()
    • LuminescenceWithoutMaskMetric
      • LuminescenceWithoutMaskMetric.calculate()
  • pixaris.metrics.prompts module
  • pixaris.metrics.saturation module
    • SaturationComparisonByMaskMetric
      • SaturationComparisonByMaskMetric.calculate()
    • SaturationWithoutMaskMetric
      • SaturationWithoutMaskMetric.calculate()
  • pixaris.metrics.utils module
    • dict_mean()
    • normalize_image()
  • Module contents
• pixaris.orchestration package
  • Submodules
  • pixaris.orchestration.base module
    • generate_image()
    • generate_images_based_on_dataset()
    • generate_images_for_hyperparameter_search_based_on_dataset()
  • pixaris.orchestration.kubernetes module
    • copy_bytes_to_pod()
    • pixaris_orchestration_kubernetes_locally()
  • Module contents
• pixaris.utils package
  • Submodules
  • pixaris.utils.bigquery module
    • create_schema_from_dict()
    • ensure_table_exists()
    • python_type_to_bq_type()
  • pixaris.utils.hyperparameters module
    • expand_hyperparameters()
    • generate_hyperparameter_grid()
  • pixaris.utils.merge_dicts module
    • merge_dicts()
  • pixaris.utils.retry module
    • retry()
  • Module contents

Module contents