Index

_

__call__() (fruit_project.utils.early_stop.EarlyStopping method) __getitem__() (fruit_project.utils.datasets.cls_dataset.CLS_DS method) (fruit_project.utils.datasets.det_dataset.DET_DS method), [1]

__len__() (fruit_project.utils.datasets.cls_dataset.CLS_DS method) (fruit_project.utils.datasets.det_dataset.DET_DS method), [1] _load_checkpoint() (fruit_project.utils.trainer.Trainer method) _save_checkpoint() (fruit_project.utils.trainer.Trainer method)

A

accum_steps (fruit_project.utils.trainer.Trainer attribute)

B

best_metric (fruit_project.utils.early_stop.EarlyStopping attribute)

C

cfg (fruit_project.utils.early_stop.EarlyStopping attribute) (fruit_project.utils.trainer.Trainer attribute) cleanup_checkpoints() (fruit_project.utils.early_stop.EarlyStopping method) CLS_DS (class in fruit_project.utils.datasets.cls_dataset) collate_fn() (in module fruit_project.utils.data)

conf (fruit_project.utils.metrics.ConfusionMatrix attribute) config_dir (fruit_project.utils.datasets.det_dataset.DET_DS attribute) config_file (fruit_project.utils.datasets.det_dataset.DET_DS attribute) ConfusionMatrix (class in fruit_project.utils.metrics) counter (fruit_project.utils.early_stop.EarlyStopping attribute)

D

delta (fruit_project.utils.early_stop.EarlyStopping attribute) DET_DS (class in fruit_project.utils.datasets.det_dataset)

device (fruit_project.utils.trainer.Trainer attribute) download_dataset() (in module fruit_project.utils.data)

E

early_stopping (fruit_project.utils.trainer.Trainer attribute) earlystop (fruit_project.utils.early_stop.EarlyStopping attribute)

EarlyStopping (class in fruit_project.utils.early_stop) eps (fruit_project.utils.metrics.ConfusionMatrix attribute) eval() (fruit_project.utils.trainer.Trainer method)

F

fit() (fruit_project.utils.trainer.Trainer method) format_targets_for_map() (fruit_project.utils.trainer.Trainer method) fruit_project module fruit_project.main module fruit_project.models module fruit_project.models.model_factory module fruit_project.models.transforms_factory module fruit_project.utils module fruit_project.utils.data module

fruit_project.utils.datasets module fruit_project.utils.datasets.cls_dataset module fruit_project.utils.datasets.det_dataset module fruit_project.utils.early_stop module fruit_project.utils.general module fruit_project.utils.logging module fruit_project.utils.metrics module fruit_project.utils.trainer module

G

get_best_model() (fruit_project.utils.early_stop.EarlyStopping method) get_epoch_log_data() (fruit_project.utils.trainer.Trainer method) get_labels_and_mappings() (in module fruit_project.utils.data) get_matrix() (fruit_project.utils.metrics.ConfusionMatrix method) get_model() (in module fruit_project.models.model_factory) get_optimizer() (fruit_project.utils.trainer.Trainer method)

get_RTDETRv2() (in module fruit_project.models.model_factory) get_run_name() (in module fruit_project.utils.logging) get_sampler() (in module fruit_project.utils.data) get_scheduler() (fruit_project.utils.trainer.Trainer method) get_transforms() (in module fruit_project.models.transforms_factory) get_val_log_data() (fruit_project.utils.trainer.Trainer method)

I

id2lbl (fruit_project.utils.datasets.cls_dataset.CLS_DS attribute) (fruit_project.utils.datasets.det_dataset.DET_DS attribute), [1] image_dir (fruit_project.utils.datasets.det_dataset.DET_DS attribute), [1] image_paths (fruit_project.utils.datasets.det_dataset.DET_DS attribute), [1]

initwandb() (in module fruit_project.utils.logging) input_size (fruit_project.utils.datasets.det_dataset.DET_DS attribute), [1] iou_thres (fruit_project.utils.metrics.ConfusionMatrix attribute) is_hf_model() (in module fruit_project.utils.general)

L

label_dir (fruit_project.utils.datasets.det_dataset.DET_DS attribute), [1] labels (fruit_project.utils.datasets.cls_dataset.CLS_DS attribute) (fruit_project.utils.datasets.det_dataset.DET_DS attribute), [1] lbl2id (fruit_project.utils.datasets.cls_dataset.CLS_DS attribute) (fruit_project.utils.datasets.det_dataset.DET_DS attribute), [1] log_checkpoint_artifact() (in module fruit_project.utils.logging)

log_class_value_counts() (in module fruit_project.utils.logging) log_detection_confusion_matrix() (in module fruit_project.utils.logging) log_images() (in module fruit_project.utils.logging) log_model_params() (in module fruit_project.utils.logging) log_training_time() (in module fruit_project.utils.logging) log_transforms() (in module fruit_project.utils.logging)

M

main() (in module fruit_project.main) make_dataloaders() (in module fruit_project.utils.data) make_datasets() (in module fruit_project.utils.data) matrix (fruit_project.utils.metrics.ConfusionMatrix attribute) model (fruit_project.utils.trainer.Trainer attribute) module fruit_project fruit_project.main fruit_project.models fruit_project.models.model_factory fruit_project.models.transforms_factory fruit_project.utils fruit_project.utils.data fruit_project.utils.datasets fruit_project.utils.datasets.cls_dataset fruit_project.utils.datasets.det_dataset fruit_project.utils.early_stop fruit_project.utils.general fruit_project.utils.logging fruit_project.utils.metrics fruit_project.utils.trainer

move_labels_to_device() (fruit_project.utils.trainer.Trainer method)

N

name (fruit_project.utils.early_stop.EarlyStopping attribute) (fruit_project.utils.trainer.Trainer attribute)

nc (fruit_project.utils.metrics.ConfusionMatrix attribute) nested_to_cpu() (fruit_project.utils.trainer.Trainer method)

O

optimizer (fruit_project.utils.trainer.Trainer attribute)

P

path (fruit_project.utils.early_stop.EarlyStopping attribute) patience (fruit_project.utils.early_stop.EarlyStopping attribute) plot() (fruit_project.utils.metrics.ConfusionMatrix method)

plot_img() (in module fruit_project.utils.general) process_batch() (fruit_project.utils.metrics.ConfusionMatrix method) processor (fruit_project.utils.trainer.Trainer attribute)

R

rlimit (in module fruit_project.utils.data) root_dir (fruit_project.utils.datasets.det_dataset.DET_DS attribute), [1]

run (fruit_project.utils.early_stop.EarlyStopping attribute) (fruit_project.utils.trainer.Trainer attribute)

S

samples (fruit_project.utils.datasets.cls_dataset.CLS_DS attribute) save_model() (fruit_project.utils.early_stop.EarlyStopping method) saved_checkpoints (fruit_project.utils.early_stop.EarlyStopping attribute) scaler (fruit_project.utils.trainer.Trainer attribute) scheduler (fruit_project.utils.trainer.Trainer attribute)

seed_worker() (in module fruit_project.utils.general) set_seed() (in module fruit_project.utils.general) set_transforms() (in module fruit_project.utils.data) start_epoch (fruit_project.utils.trainer.Trainer attribute) supported_models (in module fruit_project.models.model_factory)

T

test_dl (fruit_project.utils.trainer.Trainer attribute) train() (fruit_project.utils.trainer.Trainer method) train_dl (fruit_project.utils.trainer.Trainer attribute)

Trainer (class in fruit_project.utils.trainer) transforms (fruit_project.utils.datasets.cls_dataset.CLS_DS attribute) (fruit_project.utils.datasets.det_dataset.DET_DS attribute), [1] type (fruit_project.utils.datasets.det_dataset.DET_DS attribute), [1]

U

unnormalize() (in module fruit_project.utils.general)

V

val_dl (fruit_project.utils.trainer.Trainer attribute)