lit_ecology_classifier
latest
Contents:
Quickstart Guide
Introduction to PyTorch Lightning
ArgumentParser
SLURM Introduction and Job Submission Script
lit_ecology_classifier
lit_ecology_classifier
Index
Edit on GitHub
Index
A
|
B
|
C
|
D
|
F
|
G
|
I
|
L
|
M
|
O
|
P
|
S
|
T
|
V
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)