Index

_

__call__() (ucs.data.transform.Augmentation method)

A

alpha (ucs.utils.config.TrainingConfig attribute)

apply_color_map() (in module ucs.utils) Augmentation (class in ucs.data.transform)

B

BaseError

batch_size (ucs.utils.config.DatasetConfig attribute)

C

calculate_confusion_matrix() (ucs.model.lightning_model.SegformerFinetuner method) callbacks (ucs.utils.config.Config attribute) CallbacksConfig (class in ucs.utils.config) CeDiceLoss (class in ucs.utils.metrics) CheckpointDirectoryError

CheckpointNotFoundError checkpoints (ucs.utils.config.DirectoriesConfig attribute) compute_class_weights() (in module ucs.utils.metrics) Config (class in ucs.utils.config) ConfigId2LabelError configure_optimizers() (ucs.model.lightning_model.SegformerFinetuner method)

D

dataset (ucs.utils.config.Config attribute) dataset_path (ucs.utils.config.DatasetConfig attribute) DatasetConfig (class in ucs.utils.config)

DiceLoss (class in ucs.utils.metrics) directories (ucs.utils.config.Config attribute) DirectoriesConfig (class in ucs.utils.config) do_reduce_labels (ucs.utils.config.DatasetConfig attribute)

E

early_stop_mode (ucs.utils.config.CallbacksConfig attribute) early_stop_monitor (ucs.utils.config.CallbacksConfig attribute) early_stop_patience (ucs.utils.config.CallbacksConfig attribute) exit_code (ucs.core.errors.BaseError attribute) (ucs.core.errors.CheckpointDirectoryError attribute) (ucs.core.errors.CheckpointNotFoundError attribute) (ucs.core.errors.ConfigId2LabelError attribute) (ucs.core.errors.LossIgnoreIndexError attribute) (ucs.core.errors.LossWeightsSizeError attribute) (ucs.core.errors.LossWeightsTypeError attribute) (ucs.core.errors.MultipleCheckpointsError attribute) (ucs.core.errors.NormalizeError attribute)

F

find_checkpoint() (in module ucs.utils) FocalLoss (class in ucs.utils.metrics) format_metrics() (in module ucs.utils) forward() (ucs.model.lightning_model.SegformerFinetuner method) (ucs.utils.metrics.CeDiceLoss method) (ucs.utils.metrics.DiceLoss method) (ucs.utils.metrics.FocalLoss method)

freeze_encoder_layers() (ucs.model.lightning_model.SegformerFinetuner method)

G

get_last_version() (in module ucs.utils)

get_next_version() (in module ucs.utils)

H

handle() (ucs.core.errors.BaseError method)

I

id2label (ucs.utils.config.TrainingConfig attribute)

ignore_index (ucs.utils.config.TrainingConfig attribute) initialize_weights() (ucs.utils.metrics.CeDiceLoss method)

L

learning_rate (ucs.utils.config.TrainingConfig attribute) load_class_weights() (in module ucs.utils) load_config() (ucs.utils.config.Config class method) log_step() (ucs.model.lightning_model.SegformerFinetuner method)

logs (ucs.utils.config.DirectoriesConfig attribute) LossAdjustmentCallback (class in ucs.utils.callbacks) LossIgnoreIndexError LossWeightsSizeError LossWeightsTypeError

M

max_epochs (ucs.utils.config.TrainingConfig attribute) message (ucs.core.errors.BaseError attribute) (ucs.core.errors.CheckpointDirectoryError attribute) (ucs.core.errors.CheckpointNotFoundError attribute) (ucs.core.errors.ConfigId2LabelError attribute) (ucs.core.errors.LossIgnoreIndexError attribute) (ucs.core.errors.LossWeightsSizeError attribute) (ucs.core.errors.LossWeightsTypeError attribute) (ucs.core.errors.MultipleCheckpointsError attribute) (ucs.core.errors.NormalizeError attribute) model_name (ucs.utils.config.DatasetConfig attribute) (ucs.utils.config.TrainingConfig attribute)

models (ucs.utils.config.DirectoriesConfig attribute) module ucs.core.errors ucs.data.data_module ucs.data.dataset ucs.data.transform ucs.model.lightning_model ucs.utils ucs.utils.callbacks ucs.utils.metrics MultipleCheckpointsError

N

NormalizeError

num_workers (ucs.utils.config.DatasetConfig attribute)

O

on_fit_start() (ucs.model.lightning_model.SegformerFinetuner method) on_test_epoch_end() (ucs.model.lightning_model.SegformerFinetuner method) on_test_start() (ucs.model.lightning_model.SegformerFinetuner method) on_train_epoch_end() (ucs.model.lightning_model.SegformerFinetuner method)

on_train_epoch_start() (ucs.utils.callbacks.LossAdjustmentCallback method) on_train_start() (ucs.model.lightning_model.SegformerFinetuner method) on_validation_end() (ucs.utils.callbacks.UnfreezeOnPlateau method) on_validation_epoch_end() (ucs.model.lightning_model.SegformerFinetuner method)

P

pin_memory (ucs.utils.config.DatasetConfig attribute)

plot_image_and_mask() (in module ucs.utils) pretrained (ucs.utils.config.DirectoriesConfig attribute)

R

results (ucs.utils.config.DirectoriesConfig attribute)

S

save_class_weights() (in module ucs.utils) save_confusion_matrix_plot() (in module ucs.utils) save_model_mode (ucs.utils.config.CallbacksConfig attribute) save_model_monitor (ucs.utils.config.CallbacksConfig attribute) save_pretrained_model() (ucs.model.lightning_model.SegformerFinetuner method) SaveModel (class in ucs.utils.callbacks)

SegformerFinetuner (class in ucs.model.lightning_model) SegmentationDataModule (class in ucs.data.data_module) SegMetrics (class in ucs.utils.metrics) SemanticSegmentationDataset (class in ucs.data.dataset) set_stage() (ucs.utils.metrics.CeDiceLoss method) step() (ucs.model.lightning_model.SegformerFinetuner method)

T

test_step() (ucs.model.lightning_model.SegformerFinetuner method) TestMetrics (class in ucs.utils.metrics)

training (ucs.utils.config.Config attribute) training_step() (ucs.model.lightning_model.SegformerFinetuner method) TrainingConfig (class in ucs.utils.config)

U

ucs.core.errors module ucs.data.data_module module ucs.data.dataset module ucs.data.transform module ucs.model.lightning_model module

ucs.utils module ucs.utils.callbacks module ucs.utils.metrics module unfreeze_encoder_layers() (ucs.model.lightning_model.SegformerFinetuner method) UnfreezeOnPlateau (class in ucs.utils.callbacks) update() (ucs.utils.metrics.SegMetrics method)

V

validation_step() (ucs.model.lightning_model.SegformerFinetuner method)

W

weight_decay (ucs.utils.config.TrainingConfig attribute)

weighting_strategy (ucs.utils.config.TrainingConfig attribute)