lit_ecology_classifier package

Subpackages

• lit_ecology_classifier.data package
  • Submodules
  • lit_ecology_classifier.data.datamodule module
    • DataModule
      • DataModule.tarpath
      • DataModule.batch_size
      • DataModule.dataset
      • DataModule.testing
      • DataModule.priority_classes
      • DataModule.splits
      • DataModule.predict_dataloader()
      • DataModule.setup()
      • DataModule.test_dataloader()
      • DataModule.train_dataloader()
      • DataModule.val_dataloader()
  • lit_ecology_classifier.data.tardataset module
    • TarImageDataset
      • TarImageDataset.tar_path
      • TarImageDataset.class_map_path
      • TarImageDataset.priority_classes
      • TarImageDataset.train
      • TarImageDataset.TTA
      • TarImageDataset.get_label_from_filename()
      • TarImageDataset.shuffle()
  • Module contents
• lit_ecology_classifier.helpers package
  • Submodules
  • lit_ecology_classifier.helpers.argparser module
    • argparser()
    • inference_argparser()
  • lit_ecology_classifier.helpers.calc_class_weights module
    • calculate_class_weights()
  • lit_ecology_classifier.helpers.helpers module
    • CosineWarmupScheduler
      • CosineWarmupScheduler.get_lr()
      • CosineWarmupScheduler.get_lr_factor()
      • CosineWarmupScheduler.get_lr()
      • CosineWarmupScheduler.get_lr_factor()
    • FocalLoss
      • FocalLoss.forward()
    • TTA_collate_fn()
    • cvd_colormap()
    • define_priority_classes()
    • define_rest_classes()
    • gmean()
    • output_results()
    • plot_confusion_matrix()
    • plot_loss_acc()
    • plot_reduced_classes()
    • plot_score_distributions()
    • setup_callbacks()
    • setup_classmap()
  • Module contents
• lit_ecology_classifier.models package
  • Submodules
  • lit_ecology_classifier.models.model module
    • LitClassifier
      • LitClassifier.TTA()
      • LitClassifier.configure_optimizers()
      • LitClassifier.forward()
      • LitClassifier.load_datamodule()
      • LitClassifier.on_fit_end()
      • LitClassifier.on_predict_epoch_end()
      • LitClassifier.on_predict_start()
      • LitClassifier.on_test_epoch_end()
      • LitClassifier.on_test_epoch_start()
      • LitClassifier.on_validation_epoch_end()
      • LitClassifier.on_validation_epoch_start()
      • LitClassifier.predict_step()
      • LitClassifier.test_step()
      • LitClassifier.training_step()
      • LitClassifier.validation_step()
  • lit_ecology_classifier.models.setup_model module
    • set_trainable_params()
    • setup_model()
  • Module contents

Submodules

lit_ecology_classifier.main module

lit_ecology_classifier.predict module

Module contents