LICENSE
README.md
pyproject.toml
luxonis_train/__init__.py
luxonis_train/__main__.py
luxonis_train/registry.py
luxonis_train/tasks.py
luxonis_train/typing.py
luxonis_train/upgrade.py
luxonis_train/variants.py
luxonis_train.egg-info/PKG-INFO
luxonis_train.egg-info/SOURCES.txt
luxonis_train.egg-info/dependency_links.txt
luxonis_train.egg-info/entry_points.txt
luxonis_train.egg-info/requires.txt
luxonis_train.egg-info/top_level.txt
luxonis_train/assigners/__init__.py
luxonis_train/assigners/atss_assigner.py
luxonis_train/assigners/tal_assigner.py
luxonis_train/assigners/utils.py
luxonis_train/attached_modules/__init__.py
luxonis_train/attached_modules/base_attached_module.py
luxonis_train/attached_modules/losses/__init__.py
luxonis_train/attached_modules/losses/adaptive_detection_loss.py
luxonis_train/attached_modules/losses/base_loss.py
luxonis_train/attached_modules/losses/bce_with_logits.py
luxonis_train/attached_modules/losses/cross_entropy.py
luxonis_train/attached_modules/losses/ctc_loss.py
luxonis_train/attached_modules/losses/efficient_keypoint_bbox_loss.py
luxonis_train/attached_modules/losses/embedding_losses.py
luxonis_train/attached_modules/losses/fomo_localization_loss.py
luxonis_train/attached_modules/losses/ohem_loss.py
luxonis_train/attached_modules/losses/precision_dfl_detection_loss.py
luxonis_train/attached_modules/losses/precision_dfl_segmentation_loss.py
luxonis_train/attached_modules/losses/reconstruction_segmentation_loss.py
luxonis_train/attached_modules/losses/sigmoid_focal_loss.py
luxonis_train/attached_modules/losses/smooth_bce_with_logits.py
luxonis_train/attached_modules/losses/softmax_focal_loss.py
luxonis_train/attached_modules/metrics/__init__.py
luxonis_train/attached_modules/metrics/base_metric.py
luxonis_train/attached_modules/metrics/dice_coefficient.py
luxonis_train/attached_modules/metrics/embedding_metrics.py
luxonis_train/attached_modules/metrics/mean_iou.py
luxonis_train/attached_modules/metrics/object_keypoint_similarity.py
luxonis_train/attached_modules/metrics/ocr_accuracy.py
luxonis_train/attached_modules/metrics/precision_recall_curve.py
luxonis_train/attached_modules/metrics/torchmetrics.py
luxonis_train/attached_modules/metrics/utils.py
luxonis_train/attached_modules/metrics/confusion_matrix/__init__.py
luxonis_train/attached_modules/metrics/confusion_matrix/confusion_matrix.py
luxonis_train/attached_modules/metrics/confusion_matrix/detection_confusion_matrix.py
luxonis_train/attached_modules/metrics/confusion_matrix/fomo_confusion_matrix.py
luxonis_train/attached_modules/metrics/confusion_matrix/instance_segmentation_confusion_matrix.py
luxonis_train/attached_modules/metrics/confusion_matrix/recognition_confusion_matrix.py
luxonis_train/attached_modules/metrics/confusion_matrix/utils.py
luxonis_train/attached_modules/metrics/mean_average_precision/__init__.py
luxonis_train/attached_modules/metrics/mean_average_precision/mean_average_precision.py
luxonis_train/attached_modules/metrics/mean_average_precision/mean_average_precision_bbox.py
luxonis_train/attached_modules/metrics/mean_average_precision/mean_average_precision_keypoints.py
luxonis_train/attached_modules/metrics/mean_average_precision/mean_average_precision_segmentation.py
luxonis_train/attached_modules/metrics/mean_average_precision/utils.py
luxonis_train/attached_modules/visualizers/__init__.py
luxonis_train/attached_modules/visualizers/base_visualizer.py
luxonis_train/attached_modules/visualizers/bbox_visualizer.py
luxonis_train/attached_modules/visualizers/classification_visualizer.py
luxonis_train/attached_modules/visualizers/embeddings_visualizer.py
luxonis_train/attached_modules/visualizers/fomo_visualizer.py
luxonis_train/attached_modules/visualizers/instance_seg_keypoint_visualizer.py
luxonis_train/attached_modules/visualizers/instance_segmentation_visualizer.py
luxonis_train/attached_modules/visualizers/keypoint_visualizer.py
luxonis_train/attached_modules/visualizers/ocr_visualizer.py
luxonis_train/attached_modules/visualizers/segmentation_visualizer.py
luxonis_train/attached_modules/visualizers/utils.py
luxonis_train/callbacks/__init__.py
luxonis_train/callbacks/aimet_callback.py
luxonis_train/callbacks/archive_on_train_end.py
luxonis_train/callbacks/convert_on_train_end.py
luxonis_train/callbacks/ema.py
luxonis_train/callbacks/export_on_train_end.py
luxonis_train/callbacks/fail_on_no_train_batches.py
luxonis_train/callbacks/gpu_stats_monitor.py
luxonis_train/callbacks/graceful_interrupt.py
luxonis_train/callbacks/gradcam_visualizer.py
luxonis_train/callbacks/luxonis_model_summary.py
luxonis_train/callbacks/luxonis_progress_bar.py
luxonis_train/callbacks/metadata_logger.py
luxonis_train/callbacks/needs_checkpoint.py
luxonis_train/callbacks/test_on_train_end.py
luxonis_train/callbacks/training_manager.py
luxonis_train/callbacks/training_progress_callback.py
luxonis_train/callbacks/upload_checkpoint.py
luxonis_train/config/__init__.py
luxonis_train/config/config.py
luxonis_train/config/predefined.py
luxonis_train/config/predefined_versions.py
luxonis_train/config/predefined_models/__init__.py
luxonis_train/config/predefined_models/base_predefined_model.py
luxonis_train/config/predefined_models/anomaly_detection/__init__.py
luxonis_train/config/predefined_models/anomaly_detection/v1/__init__.py
luxonis_train/config/predefined_models/anomaly_detection/v1/model.py
luxonis_train/config/predefined_models/classification/__init__.py
luxonis_train/config/predefined_models/classification/v1/__init__.py
luxonis_train/config/predefined_models/classification/v1/model.py
luxonis_train/config/predefined_models/detection/__init__.py
luxonis_train/config/predefined_models/detection/v1/__init__.py
luxonis_train/config/predefined_models/detection/v1/model.py
luxonis_train/config/predefined_models/embeddings/__init__.py
luxonis_train/config/predefined_models/embeddings/v1/__init__.py
luxonis_train/config/predefined_models/embeddings/v1/model.py
luxonis_train/config/predefined_models/fomo/__init__.py
luxonis_train/config/predefined_models/fomo/v1/__init__.py
luxonis_train/config/predefined_models/fomo/v1/model.py
luxonis_train/config/predefined_models/instance_segmentation/__init__.py
luxonis_train/config/predefined_models/instance_segmentation/v1/__init__.py
luxonis_train/config/predefined_models/instance_segmentation/v1/model.py
luxonis_train/config/predefined_models/keypoint_detection/__init__.py
luxonis_train/config/predefined_models/keypoint_detection/v1/__init__.py
luxonis_train/config/predefined_models/keypoint_detection/v1/model.py
luxonis_train/config/predefined_models/ocr_recognition/__init__.py
luxonis_train/config/predefined_models/ocr_recognition/v1/__init__.py
luxonis_train/config/predefined_models/ocr_recognition/v1/model.py
luxonis_train/config/predefined_models/segmentation/__init__.py
luxonis_train/config/predefined_models/segmentation/v1/__init__.py
luxonis_train/config/predefined_models/segmentation/v1/model.py
luxonis_train/configs/__init__.py
luxonis_train/configs/anomaly_detection_model.yaml
luxonis_train/configs/classification_heavy_model.yaml
luxonis_train/configs/classification_light_model.yaml
luxonis_train/configs/complex_model.yaml
luxonis_train/configs/defaults.yaml
luxonis_train/configs/detection_heavy_model.yaml
luxonis_train/configs/detection_light_model.yaml
luxonis_train/configs/embeddings_model.yaml
luxonis_train/configs/example_export.yaml
luxonis_train/configs/example_tuning.yaml
luxonis_train/configs/fomo_heavy_model.yaml
luxonis_train/configs/fomo_light_model.yaml
luxonis_train/configs/instance_segmentation_heavy_model.yaml
luxonis_train/configs/instance_segmentation_light_model.yaml
luxonis_train/configs/keypoint_bbox_heavy_model.yaml
luxonis_train/configs/keypoint_bbox_light_model.yaml
luxonis_train/configs/ocr_recognition_light_model.yaml
luxonis_train/configs/segmentation_heavy_model.yaml
luxonis_train/configs/segmentation_light_model.yaml
luxonis_train/core/__init__.py
luxonis_train/core/core.py
luxonis_train/core/utils/__init__.py
luxonis_train/core/utils/aimet_utils.py
luxonis_train/core/utils/annotate_utils.py
luxonis_train/core/utils/archive_utils.py
luxonis_train/core/utils/export_utils.py
luxonis_train/core/utils/infer_utils.py
luxonis_train/core/utils/train_utils.py
luxonis_train/core/utils/tune_utils.py
luxonis_train/lightning/__init__.py
luxonis_train/lightning/freezing.py
luxonis_train/lightning/luxonis_lightning.py
luxonis_train/lightning/luxonis_output.py
luxonis_train/lightning/training_plan.py
luxonis_train/lightning/utils.py
luxonis_train/loaders/__init__.py
luxonis_train/loaders/base_loader.py
luxonis_train/loaders/dummy_loader.py
luxonis_train/loaders/luxonis_loader_torch.py
luxonis_train/loaders/luxonis_perlin_loader_torch.py
luxonis_train/loaders/perlin.py
luxonis_train/nodes/__init__.py
luxonis_train/nodes/base_node.py
luxonis_train/nodes/backbones/__init__.py
luxonis_train/nodes/backbones/contextspatial.py
luxonis_train/nodes/backbones/efficientnet.py
luxonis_train/nodes/backbones/mobilenetv2.py
luxonis_train/nodes/backbones/resnet.py
luxonis_train/nodes/backbones/rexnetv1.py
luxonis_train/nodes/backbones/ddrnet/__init__.py
luxonis_train/nodes/backbones/ddrnet/blocks.py
luxonis_train/nodes/backbones/ddrnet/ddrnet.py
luxonis_train/nodes/backbones/dinov3/__init__.py
luxonis_train/nodes/backbones/dinov3/dinov3.py
luxonis_train/nodes/backbones/dinov3/rope_position_encoding.py
luxonis_train/nodes/backbones/efficientrep/__init__.py
luxonis_train/nodes/backbones/efficientrep/efficientrep.py
luxonis_train/nodes/backbones/efficientvit/__init__.py
luxonis_train/nodes/backbones/efficientvit/blocks.py
luxonis_train/nodes/backbones/efficientvit/efficientvit.py
luxonis_train/nodes/backbones/ghostfacenet/__init__.py
luxonis_train/nodes/backbones/ghostfacenet/blocks.py
luxonis_train/nodes/backbones/ghostfacenet/ghostfacenet.py
luxonis_train/nodes/backbones/micronet/__init__.py
luxonis_train/nodes/backbones/micronet/blocks.py
luxonis_train/nodes/backbones/micronet/micronet.py
luxonis_train/nodes/backbones/mobileone/__init__.py
luxonis_train/nodes/backbones/mobileone/mobileone.py
luxonis_train/nodes/backbones/pplcnet_v3/__init__.py
luxonis_train/nodes/backbones/pplcnet_v3/blocks.py
luxonis_train/nodes/backbones/pplcnet_v3/pplcnet_v3.py
luxonis_train/nodes/backbones/recsubnet/__init__.py
luxonis_train/nodes/backbones/recsubnet/recsubnet.py
luxonis_train/nodes/backbones/repvgg/__init__.py
luxonis_train/nodes/backbones/repvgg/repvgg.py
luxonis_train/nodes/blocks/__init__.py
luxonis_train/nodes/blocks/blocks.py
luxonis_train/nodes/blocks/reparametrizable.py
luxonis_train/nodes/blocks/resnet.py
luxonis_train/nodes/blocks/unet.py
luxonis_train/nodes/blocks/utils.py
luxonis_train/nodes/heads/__init__.py
luxonis_train/nodes/heads/base_detection_head.py
luxonis_train/nodes/heads/base_head.py
luxonis_train/nodes/heads/bisenet_head.py
luxonis_train/nodes/heads/classification_head.py
luxonis_train/nodes/heads/ddrnet_segmentation_head.py
luxonis_train/nodes/heads/efficient_bbox_head.py
luxonis_train/nodes/heads/efficient_keypoint_bbox_head.py
luxonis_train/nodes/heads/fomo_head.py
luxonis_train/nodes/heads/ghostfacenet_head.py
luxonis_train/nodes/heads/ocr_ctc_head.py
luxonis_train/nodes/heads/precision_bbox_head.py
luxonis_train/nodes/heads/precision_seg_bbox_head.py
luxonis_train/nodes/heads/segmentation_head.py
luxonis_train/nodes/heads/transformer_classification_head.py
luxonis_train/nodes/heads/transformer_segmentation_head.py
luxonis_train/nodes/heads/discsubnet_head/__init__.py
luxonis_train/nodes/heads/discsubnet_head/discsubnet_head.py
luxonis_train/nodes/necks/__init__.py
luxonis_train/nodes/necks/reppan_neck/__init__.py
luxonis_train/nodes/necks/reppan_neck/blocks.py
luxonis_train/nodes/necks/reppan_neck/reppan_neck.py
luxonis_train/nodes/necks/svtr_neck/__init__.py
luxonis_train/nodes/necks/svtr_neck/blocks.py
luxonis_train/nodes/necks/svtr_neck/svtr_neck.py
luxonis_train/optimizers/__init__.py
luxonis_train/optimizers/composite_optimizer.py
luxonis_train/optimizers/optimizers.py
luxonis_train/schedulers/__init__.py
luxonis_train/schedulers/composite_scheduler.py
luxonis_train/schedulers/schedulers.py
luxonis_train/strategies/__init__.py
luxonis_train/strategies/base_strategy.py
luxonis_train/strategies/legacy.py
luxonis_train/strategies/triple_lr_sgd.py
luxonis_train/utils/__init__.py
luxonis_train/utils/annotation.py
luxonis_train/utils/boundingbox.py
luxonis_train/utils/checkpoint.py
luxonis_train/utils/dataset_metadata.py
luxonis_train/utils/exceptions.py
luxonis_train/utils/general.py
luxonis_train/utils/keypoints.py
luxonis_train/utils/logging.py
luxonis_train/utils/ocr.py
luxonis_train/utils/segmentation.py
luxonis_train/utils/spatial_transforms.py
luxonis_train/utils/tracker.py