Index

A

augment_dataset() (in module src.pipeline.augmentation)

augment_images() (in module src.pipeline.augmentation)

B

build_augmentation_transform() (in module src.pipeline.augmentation)

build_optimizer_and_scheduler() (in module src.pipeline.distillation.distillation)

C

calculate_gradient_norm() (in module src.pipeline.distillation.distillation) clean_pipeline_workspace() (in module src.pipeline.clean_pipeline) compute_iou() (in module src.pipeline.prelabelling.matching)

create_data_yaml() (in module src.utils) create_distill_yaml() (in module src.utils) create_quantize_yaml() (in module src.utils)

D

detect_device() (in module src.utils)

draw_yolo_bboxes() (in module src.utils)

E

export_versioned_results() (in module src.pipeline.human_intervention)

F

find_latest_model() (in module src.pipeline.train)

fp16_quantization() (in module src.pipeline.quantization) freeze_layers() (in module src.pipeline.distillation.distillation)

G

generate_gd_prelabelling() (in module src.pipeline.prelabelling.grounding_dino_prelabelling)

generate_yolo_prelabelling() (in module src.pipeline.prelabelling.yolo_prelabelling)

H

head_features_decoder() (in module src.pipeline.distillation.distillation)

I

import_tasks_to_project() (in module src.pipeline.human_intervention)

imx_quantization() (in module src.pipeline.quantization)

L

load_checkpoint() (in module src.pipeline.distillation.distillation) load_config() (in module src.utils)

load_models() (in module src.pipeline.distillation.distillation) load_train_config() (in module src.pipeline.train) log_training_metrics() (in module src.pipeline.distillation.distillation)

M

match_and_filter() (in module src.pipeline.prelabelling.matching) match_predictions() (in module src.pipeline.prelabelling.matching) module src.pipeline.augmentation src.pipeline.clean_pipeline src.pipeline.distillation.distillation src.pipeline.human_intervention src.pipeline.prelabelling.grounding_dino_prelabelling src.pipeline.prelabelling.matching src.pipeline.prelabelling.yolo_prelabelling src.pipeline.quantization src.pipeline.train src.utils

N

normalize_class() (in module src.pipeline.prelabelling.matching)

O

onnx_quantization() (in module src.pipeline.quantization)

P

prepare_dataset() (in module src.pipeline.distillation.distillation)

prepare_quantization_data() (in module src.utils) prepare_training_data() (in module src.utils)

Q

quantize_model() (in module src.pipeline.quantization)

R

run_human_review() (in module src.pipeline.human_intervention)

S

save_checkpoint() (in module src.pipeline.distillation.distillation) save_final_model() (in module src.pipeline.distillation.distillation) setup_label_studio() (in module src.pipeline.human_intervention) src.pipeline.augmentation module src.pipeline.clean_pipeline module src.pipeline.distillation.distillation module src.pipeline.human_intervention module src.pipeline.prelabelling.grounding_dino_prelabelling module

src.pipeline.prelabelling.matching module src.pipeline.prelabelling.yolo_prelabelling module src.pipeline.quantization module src.pipeline.train module src.utils module start_distillation() (in module src.pipeline.distillation.distillation)

T

train_epoch() (in module src.pipeline.distillation.distillation) train_loop() (in module src.pipeline.distillation.distillation)

train_model() (in module src.pipeline.train) transform_reviewed_results_to_labeled() (in module src.pipeline.human_intervention)