Index

A

argparser() (in module lit_ecology_classifier.helpers.argparser)

B

batch_size (lit_ecology_classifier.data.datamodule.DataModule attribute)

C

calculate_class_weights() (in module lit_ecology_classifier.helpers.calc_class_weights) class_map_path (lit_ecology_classifier.data.tardataset.TarImageDataset attribute)

configure_optimizers() (lit_ecology_classifier.models.model.LitClassifier method) CosineWarmupScheduler (class in lit_ecology_classifier.helpers.helpers) cvd_colormap() (in module lit_ecology_classifier.helpers.helpers)

D

DataModule (class in lit_ecology_classifier.data.datamodule) dataset (lit_ecology_classifier.data.datamodule.DataModule attribute)

define_priority_classes() (in module lit_ecology_classifier.helpers.helpers) define_rest_classes() (in module lit_ecology_classifier.helpers.helpers)

F

FocalLoss (class in lit_ecology_classifier.helpers.helpers)

forward() (lit_ecology_classifier.helpers.helpers.FocalLoss method) (lit_ecology_classifier.models.model.LitClassifier method)

G

get_label_from_filename() (lit_ecology_classifier.data.tardataset.TarImageDataset method) get_lr() (lit_ecology_classifier.helpers.helpers.CosineWarmupScheduler method), [1]

get_lr_factor() (lit_ecology_classifier.helpers.helpers.CosineWarmupScheduler method), [1] gmean() (in module lit_ecology_classifier.helpers.helpers)

I

inference_argparser() (in module lit_ecology_classifier.helpers.argparser)

L

lit_ecology_classifier module lit_ecology_classifier.data module lit_ecology_classifier.data.datamodule module lit_ecology_classifier.data.tardataset module lit_ecology_classifier.helpers module lit_ecology_classifier.helpers.argparser module lit_ecology_classifier.helpers.calc_class_weights module

lit_ecology_classifier.helpers.helpers module lit_ecology_classifier.main module lit_ecology_classifier.models module lit_ecology_classifier.models.model module lit_ecology_classifier.models.setup_model module lit_ecology_classifier.predict module LitClassifier (class in lit_ecology_classifier.models.model) load_datamodule() (lit_ecology_classifier.models.model.LitClassifier method)

M

module lit_ecology_classifier lit_ecology_classifier.data lit_ecology_classifier.data.datamodule lit_ecology_classifier.data.tardataset lit_ecology_classifier.helpers lit_ecology_classifier.helpers.argparser lit_ecology_classifier.helpers.calc_class_weights lit_ecology_classifier.helpers.helpers lit_ecology_classifier.main lit_ecology_classifier.models lit_ecology_classifier.models.model lit_ecology_classifier.models.setup_model lit_ecology_classifier.predict

O

on_fit_end() (lit_ecology_classifier.models.model.LitClassifier method) on_predict_epoch_end() (lit_ecology_classifier.models.model.LitClassifier method) on_predict_start() (lit_ecology_classifier.models.model.LitClassifier method) on_test_epoch_end() (lit_ecology_classifier.models.model.LitClassifier method)

on_test_epoch_start() (lit_ecology_classifier.models.model.LitClassifier method) on_validation_epoch_end() (lit_ecology_classifier.models.model.LitClassifier method) on_validation_epoch_start() (lit_ecology_classifier.models.model.LitClassifier method) output_results() (in module lit_ecology_classifier.helpers.helpers)

P

plot_confusion_matrix() (in module lit_ecology_classifier.helpers.helpers) plot_loss_acc() (in module lit_ecology_classifier.helpers.helpers) plot_reduced_classes() (in module lit_ecology_classifier.helpers.helpers) plot_score_distributions() (in module lit_ecology_classifier.helpers.helpers)

predict_dataloader() (lit_ecology_classifier.data.datamodule.DataModule method) predict_step() (lit_ecology_classifier.models.model.LitClassifier method) priority_classes (lit_ecology_classifier.data.datamodule.DataModule attribute) (lit_ecology_classifier.data.tardataset.TarImageDataset attribute)

S

set_trainable_params() (in module lit_ecology_classifier.models.setup_model) setup() (lit_ecology_classifier.data.datamodule.DataModule method) setup_callbacks() (in module lit_ecology_classifier.helpers.helpers)

setup_classmap() (in module lit_ecology_classifier.helpers.helpers) setup_model() (in module lit_ecology_classifier.models.setup_model) shuffle() (lit_ecology_classifier.data.tardataset.TarImageDataset method) splits (lit_ecology_classifier.data.datamodule.DataModule attribute)

T

tar_path (lit_ecology_classifier.data.tardataset.TarImageDataset attribute) TarImageDataset (class in lit_ecology_classifier.data.tardataset) tarpath (lit_ecology_classifier.data.datamodule.DataModule attribute) test_dataloader() (lit_ecology_classifier.data.datamodule.DataModule method) test_step() (lit_ecology_classifier.models.model.LitClassifier method) testing (lit_ecology_classifier.data.datamodule.DataModule attribute)

train (lit_ecology_classifier.data.tardataset.TarImageDataset attribute) train_dataloader() (lit_ecology_classifier.data.datamodule.DataModule method) training_step() (lit_ecology_classifier.models.model.LitClassifier method) TTA (lit_ecology_classifier.data.tardataset.TarImageDataset attribute) TTA() (lit_ecology_classifier.models.model.LitClassifier method) TTA_collate_fn() (in module lit_ecology_classifier.helpers.helpers)

V

val_dataloader() (lit_ecology_classifier.data.datamodule.DataModule method)

validation_step() (lit_ecology_classifier.models.model.LitClassifier method)