LICENSE
README.md
pyproject.toml
requirements-dev.txt
requirements.txt
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/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/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_models/__init__.py
luxonis_train/config/predefined_models/anomaly_detection_model.py
luxonis_train/config/predefined_models/base_predefined_model.py
luxonis_train/config/predefined_models/classification_model.py
luxonis_train/config/predefined_models/detection_fomo_model.py
luxonis_train/config/predefined_models/detection_model.py
luxonis_train/config/predefined_models/instance_segmentation_model.py
luxonis_train/config/predefined_models/keypoint_detection_model.py
luxonis_train/config/predefined_models/ocr_recognition_model.py
luxonis_train/config/predefined_models/segmentation_model.py
luxonis_train/core/__init__.py
luxonis_train/core/core.py
luxonis_train/core/utils/__init__.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/luxonis_lightning.py
luxonis_train/lightning/luxonis_output.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/optimizers.py
luxonis_train/schedulers/__init__.py
luxonis_train/schedulers/schedulers.py
luxonis_train/strategies/__init__.py
luxonis_train/strategies/base_strategy.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
tests/__init__.py
tests/conftest.py
tests/integration/__init__.py
tests/integration/backbone_model_utils.py
tests/integration/test_bump_opset_version.py
tests/integration/test_callbacks.py
tests/integration/test_cli_commands.py
tests/integration/test_combinations.py
tests/integration/test_convert.py
tests/integration/test_core.py
tests/integration/test_custom_model.py
tests/integration/test_export_input_output_identifiers.py
tests/integration/test_export_unique_identifiers.py
tests/integration/test_fixed_validation_batch_limit.py
tests/integration/test_freezing.py
tests/integration/test_loader.py
tests/integration/test_mlflow_logging.py
tests/integration/test_overfit_batches.py
tests/integration/test_overfit_convergence.py
tests/integration/test_predefined_models.py
tests/integration/test_remove_on_export.py
tests/integration/test_resume_training.py
tests/integration/test_scheduler.py
tests/integration/test_smart_logging.py
tests/unittests/__init__.py
tests/unittests/test_base_attached_module.py
tests/unittests/test_base_node.py
tests/unittests/test_blocks.py
tests/unittests/test_config.py
tests/unittests/test_export_utils.py
tests/unittests/test_assigners/__init__.py
tests/unittests/test_assigners/test_atss_assigner.py
tests/unittests/test_assigners/test_tal_assigner.py
tests/unittests/test_assigners/test_utils.py
tests/unittests/test_callbacks/__init__.py
tests/unittests/test_callbacks/test_ema.py
tests/unittests/test_callbacks/test_needs_checkpoint.py
tests/unittests/test_loaders/__init__.py
tests/unittests/test_loaders/test_base_loader.py
tests/unittests/test_losses/__init__.py
tests/unittests/test_losses/test_adaptive_detection_loss.py
tests/unittests/test_losses/test_bce_with_logits_loss.py
tests/unittests/test_losses/test_cross_entropy_loss.py
tests/unittests/test_losses/test_ctc_loss.py
tests/unittests/test_losses/test_efficient_keypoint_bbox_loss.py
tests/unittests/test_losses/test_embedding_losses.py
tests/unittests/test_losses/test_fomo_localization_loss.py
tests/unittests/test_losses/test_ohem_losses.py
tests/unittests/test_losses/test_precision_dfl_detection_loss.py
tests/unittests/test_losses/test_precision_dfl_segmentation_loss.py
tests/unittests/test_losses/test_reconstruction_segmentation_loss.py
tests/unittests/test_losses/test_sigmoid_focal_loss.py
tests/unittests/test_losses/test_smooth_bce_with_logits_loss.py
tests/unittests/test_losses/test_softmax_focal_loss.py
tests/unittests/test_losses/test_utils.py
tests/unittests/test_metrics/__init__.py
tests/unittests/test_metrics/test_confusion_matrix.py
tests/unittests/test_metrics/test_embedding_metrics.py
tests/unittests/test_metrics/test_mean_average_precision.py
tests/unittests/test_metrics/test_ocr_accuracy.py
tests/unittests/test_metrics/test_oks.py
tests/unittests/test_metrics/test_segmentation_metrics.py
tests/unittests/test_metrics/test_torchmetrics.py
tests/unittests/test_metrics/test_utils.py
tests/unittests/test_strategies/__init__.py
tests/unittests/test_strategies/test_triple_lr_sgd.py
tests/unittests/test_utils/__init__.py
tests/unittests/test_utils/test_boxutils.py
tests/unittests/test_utils/test_dataset_metadata.py
tests/unittests/test_utils/test_general.py
tests/unittests/test_utils/test_infer_utils.py
tests/unittests/test_utils/test_keypoints.py
tests/unittests/test_visualizers/__init__.py
tests/unittests/test_visualizers/test_bbox_visualizer.py
tests/unittests/test_visualizers/test_classification_visualizer.py
tests/unittests/test_visualizers/test_embeddings_visualizer.py
tests/unittests/test_visualizers/test_fomo_visualizer.py
tests/unittests/test_visualizers/test_instance_seg_keypoint_visualizer.py
tests/unittests/test_visualizers/test_instance_segmentation_visualizer.py
tests/unittests/test_visualizers/test_keypoint_visualizer.py
tests/unittests/test_visualizers/test_ocr_visualizer.py
tests/unittests/test_visualizers/test_segmentation_visualizer.py