Index

_

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

_apply_fallback_transform() (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset method) _apply_mosaic_augmentation() (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset method) _build_save_dict() (fruit_project.utils.trainer.Trainer method) _load_checkpoint() (fruit_project.utils.trainer.Trainer method) _prepare_mosaic_metadata() (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset method) _run_eval() (fruit_project.utils.trainer.Trainer method) _save_checkpoint() (fruit_project.utils.trainer.Trainer method) _validate_and_clip_bbox() (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset method)

A

accum_steps (fruit_project.utils.trainer.Trainer attribute)

AlbumentationsMosaicDataset (class in fruit_project.utils.datasets.alb_mosaic_dataset)

B

bbox_params (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute)

best_metric (fruit_project.utils.early_stop.EarlyStopping attribute)

C

cfg (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) collect_image_sizes() (fruit_project.utils.metrics.MAPEvaluator method) collect_predictions() (fruit_project.utils.metrics.MAPEvaluator method) collect_targets() (fruit_project.utils.metrics.MAPEvaluator method)

conf (fruit_project.utils.metrics.ConfusionMatrix attribute) config_dir (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute) (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) create_albumentations_mosaic_dataset() (in module fruit_project.utils.datasets.alb_mosaic_dataset) current_epoch (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute)

D

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

device (fruit_project.utils.metrics.MAPEvaluator attribute) (fruit_project.utils.trainer.Trainer attribute) disable_mosaic_epochs (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset 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) easy_compose (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute)

easy_transforms (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute) ema (fruit_project.utils.trainer.Trainer attribute) eps (fruit_project.utils.metrics.ConfusionMatrix attribute) eval() (fruit_project.utils.trainer.Trainer method)

F

fit() (fruit_project.utils.trainer.Trainer method) format_for_hf_processor() (in module fruit_project.utils.datasets.det_dataset) format_targets_for_cm() (fruit_project.utils.trainer.Trainer method) freeze_weights() (in module fruit_project.models.model_factory) 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.alb_mosaic_dataset 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_averaged_precision_recall_ultralytics_style() (fruit_project.utils.metrics.MAPEvaluator method) get_bbox_params() (in module fruit_project.models.transforms_factory) get_best_model() (fruit_project.utils.early_stop.EarlyStopping method) get_hf_model() (in module fruit_project.models.model_factory) 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_optimal_f1_ultralytics_style() (fruit_project.utils.metrics.MAPEvaluator method) get_optimizer() (fruit_project.utils.trainer.Trainer method) get_per_class() (fruit_project.utils.metrics.MAPEvaluator method) get_raw_item() (fruit_project.utils.datasets.det_dataset.DET_DS method) 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)

H

hard_transforms (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute)

I

id2label (fruit_project.utils.metrics.MAPEvaluator attribute) id2lbl (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute) (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.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute) (fruit_project.utils.datasets.det_dataset.DET_DS attribute), [1]

image_processor (fruit_project.utils.metrics.MAPEvaluator attribute) initwandb() (in module fruit_project.utils.logging) input_size (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute) (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] label_paths (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute) (fruit_project.utils.datasets.det_dataset.DET_DS attribute) labels (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute) (fruit_project.utils.datasets.cls_dataset.CLS_DS attribute) (fruit_project.utils.datasets.det_dataset.DET_DS attribute), [1] lbl2id (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute) (fruit_project.utils.datasets.cls_dataset.CLS_DS attribute) (fruit_project.utils.datasets.det_dataset.DET_DS attribute), [1] log (fruit_project.utils.early_stop.EarlyStopping attribute)

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_epoch_data() (in module fruit_project.utils.logging) log_images() (in module fruit_project.utils.logging) log_model_params() (in module fruit_project.utils.logging) log_per_class_metric() (in module fruit_project.utils.logging) log_test_data() (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) map_50_metric (fruit_project.utils.metrics.MAPEvaluator attribute) map_evaluator (fruit_project.utils.trainer.Trainer attribute) map_metric (fruit_project.utils.metrics.MAPEvaluator attribute) MAPEvaluator (class in fruit_project.utils.metrics) 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.alb_mosaic_dataset 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

mosaic_compose (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute) mosaic_prob (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute) mosaic_transform (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute) 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) normalize (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute) (fruit_project.utils.datasets.det_dataset.DET_DS attribute)

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.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute) (fruit_project.utils.datasets.det_dataset.DET_DS attribute) (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) should_apply_mosaic() (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset method) start_epoch (fruit_project.utils.trainer.Trainer attribute) supported_models (in module fruit_project.models.model_factory)

T

target_size (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset attribute) test_dl (fruit_project.utils.trainer.Trainer attribute) threshold (fruit_project.utils.metrics.MAPEvaluator attribute) total_epochs (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset 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) update() (fruit_project.utils.metrics.ConfusionMatrix method)

update_epoch() (fruit_project.utils.datasets.alb_mosaic_dataset.AlbumentationsMosaicDataset method) upload (fruit_project.utils.early_stop.EarlyStopping attribute)

V

val_dl (fruit_project.utils.trainer.Trainer attribute)