Project architecture

• config module
  • DATASET_FILE_PATH
  • DEV_DEBUG
  • DEVICE
  • WANDB_PRJ_NAME_GENERATE_SAMPLES
  • WANDB_PRJ_NAME_CLASSIFICATION
  • WANDB_PRJ_NAME_TRAIN_GENERATOR
• datasets package
  • Submodules
  • datasets.eye_fundus module
  • datasets.eye_scans module
    • EyeScans
      • EyeScans.setup()
      • EyeScans.train_dataloader()
      • EyeScans.val_dataloader()
      • EyeScans.test_dataloader()
  • datasets.opthal_anonymized module
    • get_csv_dataset()
    • OpthalAnonymizedDataset
  • datasets.prepare_dataset module
    • get_patients_paths()
    • get_images_paths()
    • get_lesion_eyes_paths()
    • get_reference_eyes_paths()
    • resize_images_and_save()
    • copy_images_to_dir()
  • datasets.prepare_opthal_anonym module
  • datasets.preprocess module
    • crop_image()
    • load_dataset_csv_file()
  • Module contents
    • EyeScans
      • EyeScans.setup()
      • EyeScans.train_dataloader()
      • EyeScans.val_dataloader()
      • EyeScans.test_dataloader()
    • OpthalAnonymizedDataset
    • get_csv_dataset()
• generate_dataset_temp module
• generate_samples module
  • generate_samples()
• main module
• models package
  • Submodules
  • models.classifier module
    • ResNetClassifier
      • ResNetClassifier.forward()
      • ResNetClassifier.training_step()
      • ResNetClassifier.validation_step()
      • ResNetClassifier.on_validation_epoch_end()
      • ResNetClassifier.test_step()
      • ResNetClassifier.on_test_epoch_end()
      • ResNetClassifier.configure_optimizers()
  • models.gausian_diffusion module
    • GaussianDiffusion
      • GaussianDiffusion.image_size
      • GaussianDiffusion.channels
      • GaussianDiffusion.num_timesteps
      • GaussianDiffusion.forward()
      • GaussianDiffusion.sample()
      • GaussianDiffusion.forward()
      • GaussianDiffusion.sample()
  • Module contents
    • GaussianDiffusion
      • GaussianDiffusion.image_size
      • GaussianDiffusion.channels
      • GaussianDiffusion.num_timesteps
      • GaussianDiffusion.forward()
      • GaussianDiffusion.sample()
      • GaussianDiffusion.forward()
      • GaussianDiffusion.sample()
    • ResNetClassifier
      • ResNetClassifier.forward()
      • ResNetClassifier.training_step()
      • ResNetClassifier.validation_step()
      • ResNetClassifier.on_validation_epoch_end()
      • ResNetClassifier.test_step()
      • ResNetClassifier.on_test_epoch_end()
      • ResNetClassifier.configure_optimizers()
• pipeline package
  • Subpackages
    • pipeline.blocks package
      • Submodules
      • pipeline.blocks.blocks module
      • pipeline.blocks.data_dto module
      • pipeline.blocks.experiment_dto module
      • pipeline.blocks.general_dto module
      • pipeline.blocks.models_dto module
      • pipeline.blocks.output_dto module
      • pipeline.blocks.testing_dto module
      • Module contents
  • Submodules
  • pipeline.parser module
    • j_print()
    • read_config_file()
    • parse_config()
    • get_experiment_configs()
  • pipeline.pipeline module
  • Module contents
• run_classification_experiment module
  • main()
• run_experiment module
  • main()
• test_pipeline module
• trackers package
  • Submodules
  • trackers.callbacks module
    • ImagePredictionLogger
      • ImagePredictionLogger.on_validation_epoch_end()
  • trackers.wandb module
    • is_square()
    • WandbTracker
      • WandbTracker.observe_model()
      • WandbTracker.log()
      • WandbTracker.log_images()
      • WandbTracker.finish()
      • WandbTracker.get_experiment_name()
      • WandbTracker.save_model()
      • WandbTracker.update_step()
  • Module contents
    • WandbTracker
      • WandbTracker.observe_model()
      • WandbTracker.log()
      • WandbTracker.log_images()
      • WandbTracker.finish()
      • WandbTracker.get_experiment_name()
      • WandbTracker.save_model()
      • WandbTracker.update_step()
    • get_tracker_class()
    • ImagePredictionLogger
      • ImagePredictionLogger.on_validation_epoch_end()
• trainers package
  • Submodules
  • trainers.classifier_trainer module
  • trainers.trainer module
    • Trainer
      • Trainer.save()
      • Trainer.keep_last_models()
      • Trainer.load()
      • Trainer.train()
  • Module contents
    • Trainer
      • Trainer.save()
      • Trainer.keep_last_models()
      • Trainer.load()
      • Trainer.train()
• utils package
  • Submodules
  • utils.artifact_utilities module
    • check_results_dir_exists()
  • utils.files_managment module
    • copy_results_directory()
  • utils.transforms module
    • HorizontalCenterCrop
      • HorizontalCenterCrop.width
      • HorizontalCenterCrop.__call__()
  • Module contents
    • copy_results_directory()
• validation package
  • Submodules
  • validation.embeddings module
    • Embeddings
      • Embeddings.image_to_embedding()
      • Embeddings.get_images_embeddings()
      • Embeddings.get_images_embeddings_from_dataset()
      • Embeddings.save_embeddings()
      • Embeddings.load_embeddings()
      • Embeddings.create_umap_plot()
      • Embeddings.create_violin_plot()
      • Embeddings.calculate_embeddings_distances()
      • Embeddings.calculate_cosine_similarity()
  • validation.fid module
  • Module contents
    • Embeddings
      • Embeddings.image_to_embedding()
      • Embeddings.get_images_embeddings()
      • Embeddings.get_images_embeddings_from_dataset()
      • Embeddings.save_embeddings()
      • Embeddings.load_embeddings()
      • Embeddings.create_umap_plot()
      • Embeddings.create_violin_plot()
      • Embeddings.calculate_embeddings_distances()
      • Embeddings.calculate_cosine_similarity()