├── .devcontainer └── devcontainer.json ├── .gitignore ├── Dockerfile ├── LICENSE ├── Makefile ├── README.md ├── blog post ├── Yolo7 blog post.ipynb ├── __init__.py └── yolov7_head_arch.png ├── examples ├── evaluate_coco.py ├── minimal_finetune_cars.py └── train_cars.py ├── extract_state_dicts.py ├── get_coco.sh ├── requirements.txt ├── setup.py └── yolov7 ├── __init__.py ├── anchors.py ├── dataset.py ├── evaluation ├── __init__.py ├── calculate_map_callback.py └── coco_evaluator.py ├── loss.py ├── loss_factory.py ├── models ├── __init__.py ├── config_builder.py ├── core │ ├── __init__.py │ ├── detection_heads.py │ ├── layer_operations.py │ └── layers.py ├── model_configs.py ├── model_factory.py └── yolo.py ├── mosaic.py ├── plotting.py ├── trainer.py └── utils.py /.devcontainer/devcontainer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/.devcontainer/devcontainer.json -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/.gitignore -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/README.md -------------------------------------------------------------------------------- /blog post/Yolo7 blog post.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/blog post/Yolo7 blog post.ipynb -------------------------------------------------------------------------------- /blog post/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /blog post/yolov7_head_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/blog post/yolov7_head_arch.png -------------------------------------------------------------------------------- /examples/evaluate_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/examples/evaluate_coco.py -------------------------------------------------------------------------------- /examples/minimal_finetune_cars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/examples/minimal_finetune_cars.py -------------------------------------------------------------------------------- /examples/train_cars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/examples/train_cars.py -------------------------------------------------------------------------------- /extract_state_dicts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/extract_state_dicts.py -------------------------------------------------------------------------------- /get_coco.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/get_coco.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/setup.py -------------------------------------------------------------------------------- /yolov7/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/__init__.py -------------------------------------------------------------------------------- /yolov7/anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/anchors.py -------------------------------------------------------------------------------- /yolov7/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/dataset.py -------------------------------------------------------------------------------- /yolov7/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/evaluation/__init__.py -------------------------------------------------------------------------------- /yolov7/evaluation/calculate_map_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/evaluation/calculate_map_callback.py -------------------------------------------------------------------------------- /yolov7/evaluation/coco_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/evaluation/coco_evaluator.py -------------------------------------------------------------------------------- /yolov7/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/loss.py -------------------------------------------------------------------------------- /yolov7/loss_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/loss_factory.py -------------------------------------------------------------------------------- /yolov7/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /yolov7/models/config_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/models/config_builder.py -------------------------------------------------------------------------------- /yolov7/models/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /yolov7/models/core/detection_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/models/core/detection_heads.py -------------------------------------------------------------------------------- /yolov7/models/core/layer_operations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/models/core/layer_operations.py -------------------------------------------------------------------------------- /yolov7/models/core/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/models/core/layers.py -------------------------------------------------------------------------------- /yolov7/models/model_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/models/model_configs.py -------------------------------------------------------------------------------- /yolov7/models/model_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/models/model_factory.py -------------------------------------------------------------------------------- /yolov7/models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/models/yolo.py -------------------------------------------------------------------------------- /yolov7/mosaic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/mosaic.py -------------------------------------------------------------------------------- /yolov7/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/plotting.py -------------------------------------------------------------------------------- /yolov7/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/trainer.py -------------------------------------------------------------------------------- /yolov7/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chris-hughes10/Yolov7-training/HEAD/yolov7/utils.py --------------------------------------------------------------------------------