LICENSE
README.md
pyproject.toml
requirements-dev.txt
requirements.txt
luxonis_train/__init__.py
luxonis_train/__main__.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/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/implicit_keypoint_bbox_loss.py
luxonis_train/attached_modules/losses/keypoint_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/common.py
luxonis_train/attached_modules/metrics/mean_average_precision.py
luxonis_train/attached_modules/metrics/mean_average_precision_keypoints.py
luxonis_train/attached_modules/metrics/object_keypoint_similarity.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/keypoint_visualizer.py
luxonis_train/attached_modules/visualizers/multi_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/export_on_train_end.py
luxonis_train/callbacks/luxonis_progress_bar.py
luxonis_train/callbacks/metadata_logger.py
luxonis_train/callbacks/module_freezer.py
luxonis_train/callbacks/test_on_train_end.py
luxonis_train/callbacks/upload_checkpoint_on_train_end.py
luxonis_train/core/__init__.py
luxonis_train/core/core.py
luxonis_train/core/exporter.py
luxonis_train/core/inferer.py
luxonis_train/core/trainer.py
luxonis_train/core/tuner.py
luxonis_train/models/__init__.py
luxonis_train/models/luxonis_model.py
luxonis_train/models/luxonis_output.py
luxonis_train/models/predefined_models/__init__.py
luxonis_train/models/predefined_models/base_predefined_model.py
luxonis_train/models/predefined_models/classification_model.py
luxonis_train/models/predefined_models/detection_model.py
luxonis_train/models/predefined_models/keypoint_detection_model.py
luxonis_train/models/predefined_models/segmentation_model.py
luxonis_train/nodes/__init__.py
luxonis_train/nodes/base_node.py
luxonis_train/nodes/bisenet_head.py
luxonis_train/nodes/classification_head.py
luxonis_train/nodes/contextspatial.py
luxonis_train/nodes/efficient_bbox_head.py
luxonis_train/nodes/efficientnet.py
luxonis_train/nodes/efficientrep.py
luxonis_train/nodes/implicit_keypoint_bbox_head.py
luxonis_train/nodes/micronet.py
luxonis_train/nodes/mobilenetv2.py
luxonis_train/nodes/mobileone.py
luxonis_train/nodes/reppan_neck.py
luxonis_train/nodes/repvgg.py
luxonis_train/nodes/resnet18.py
luxonis_train/nodes/rexnetv1.py
luxonis_train/nodes/segmentation_head.py
luxonis_train/nodes/activations/__init__.py
luxonis_train/nodes/activations/activations.py
luxonis_train/nodes/blocks/__init__.py
luxonis_train/nodes/blocks/blocks.py
luxonis_train/utils/__init__.py
luxonis_train/utils/boxutils.py
luxonis_train/utils/config.py
luxonis_train/utils/general.py
luxonis_train/utils/optimizers.py
luxonis_train/utils/registry.py
luxonis_train/utils/schedulers.py
luxonis_train/utils/tracker.py
luxonis_train/utils/types.py
luxonis_train/utils/assigners/__init__.py
luxonis_train/utils/assigners/atts_assigner.py
luxonis_train/utils/assigners/tal_assigner.py
luxonis_train/utils/assigners/utils.py
luxonis_train/utils/loaders/__init__.py
luxonis_train/utils/loaders/base_loader.py
luxonis_train/utils/loaders/luxonis_loader_torch.py
tests/integration/conftest.py
tests/integration/test_sanity.py
tests/unittests/__init__.py
tests/unittests/test_losses/__init__.py
tests/unittests/test_losses/test_bce_with_logits_loss.py
tests/unittests/test_utils/__init__.py
tests/unittests/test_utils/test_boxutils.py
tests/unittests/test_utils/test_assigners/test_atts_assigner.py
tests/unittests/test_utils/test_assigners/test_tal_assigner.py
tests/unittests/test_utils/test_assigners/test_utils.py
tests/unittests/test_utils/test_loaders/test_base_loader.py
tools/main.py