├── LICENSE ├── README.md ├── cfg ├── baseline │ ├── r50-csp.yaml │ ├── x50-csp.yaml │ ├── yolor-csp-x.yaml │ ├── yolor-csp.yaml │ ├── yolor-d6.yaml │ ├── yolor-e6.yaml │ ├── yolor-p6.yaml │ ├── yolor-w6.yaml │ ├── yolov3-spp.yaml │ ├── yolov3.yaml │ └── yolov4-csp.yaml ├── deploy │ ├── yolov7-d6.yaml │ ├── yolov7-e6.yaml │ ├── yolov7-e6e.yaml │ ├── yolov7-tiny-silu.yaml │ ├── yolov7-tiny.yaml │ ├── yolov7-w6.yaml │ ├── yolov7.yaml │ └── yolov7x.yaml └── training │ ├── yolov7-d6.yaml │ ├── yolov7-e6.yaml │ ├── yolov7-e6e.yaml │ ├── yolov7-tiny.yaml │ ├── yolov7-w6.yaml │ ├── yolov7.yaml │ └── yolov7x.yaml ├── data ├── coco.yaml ├── hyp.scratch.custom.yaml ├── hyp.scratch.p5.yaml ├── hyp.scratch.p6.yaml └── hyp.scratch.tiny.yaml ├── detect.py ├── export.py ├── figure ├── horses_prediction.jpg ├── mask.png ├── performance.png ├── pose.png ├── tennis.jpg ├── tennis_caption.png ├── tennis_panoptic.png └── tennis_semantic.jpg ├── imgANDvideo ├── friends.jpg ├── girl.jpg ├── huaxue.jpg ├── myVideo.mp4 ├── train.png ├── xican.jpg └── yuanli.png ├── models ├── __init__.py ├── common.py ├── experimental.py └── yolo.py ├── scripts └── get_coco.sh ├── test.py ├── tools ├── YOLOv7-Dynamic-Batch-ONNXRUNTIME.ipynb ├── YOLOv7-Dynamic-Batch-TENSORRT.ipynb ├── YOLOv7CoreML.ipynb ├── YOLOv7onnx.ipynb ├── YOLOv7trt.ipynb ├── compare_YOLOv7_vs_YOLOv5m6.ipynb ├── compare_YOLOv7_vs_YOLOv5m6_half.ipynb ├── compare_YOLOv7_vs_YOLOv5s6.ipynb ├── compare_YOLOv7e6_vs_YOLOv5x6.ipynb ├── compare_YOLOv7e6_vs_YOLOv5x6_half.ipynb ├── instance.ipynb ├── keypoint.ipynb ├── reparameterization.ipynb └── visualization.ipynb ├── train.py └── utils ├── __init__.py ├── activations.py ├── add_nms.py ├── autoanchor.py ├── aws ├── __init__.py ├── mime.sh ├── resume.py └── userdata.sh ├── datasets.py ├── general.py ├── google_app_engine ├── Dockerfile ├── additional_requirements.txt └── app.yaml ├── google_utils.py ├── loss.py ├── metrics.py ├── plots.py ├── torch_utils.py └── wandb_logging ├── __init__.py ├── log_dataset.py └── wandb_utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/README.md -------------------------------------------------------------------------------- /cfg/baseline/r50-csp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/baseline/r50-csp.yaml -------------------------------------------------------------------------------- /cfg/baseline/x50-csp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/baseline/x50-csp.yaml -------------------------------------------------------------------------------- /cfg/baseline/yolor-csp-x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/baseline/yolor-csp-x.yaml -------------------------------------------------------------------------------- /cfg/baseline/yolor-csp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/baseline/yolor-csp.yaml -------------------------------------------------------------------------------- /cfg/baseline/yolor-d6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/baseline/yolor-d6.yaml -------------------------------------------------------------------------------- /cfg/baseline/yolor-e6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/baseline/yolor-e6.yaml -------------------------------------------------------------------------------- /cfg/baseline/yolor-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/baseline/yolor-p6.yaml -------------------------------------------------------------------------------- /cfg/baseline/yolor-w6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/baseline/yolor-w6.yaml -------------------------------------------------------------------------------- /cfg/baseline/yolov3-spp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/baseline/yolov3-spp.yaml -------------------------------------------------------------------------------- /cfg/baseline/yolov3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/baseline/yolov3.yaml -------------------------------------------------------------------------------- /cfg/baseline/yolov4-csp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/baseline/yolov4-csp.yaml -------------------------------------------------------------------------------- /cfg/deploy/yolov7-d6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/deploy/yolov7-d6.yaml -------------------------------------------------------------------------------- /cfg/deploy/yolov7-e6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/deploy/yolov7-e6.yaml -------------------------------------------------------------------------------- /cfg/deploy/yolov7-e6e.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/deploy/yolov7-e6e.yaml -------------------------------------------------------------------------------- /cfg/deploy/yolov7-tiny-silu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/deploy/yolov7-tiny-silu.yaml -------------------------------------------------------------------------------- /cfg/deploy/yolov7-tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/deploy/yolov7-tiny.yaml -------------------------------------------------------------------------------- /cfg/deploy/yolov7-w6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/deploy/yolov7-w6.yaml -------------------------------------------------------------------------------- /cfg/deploy/yolov7.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/deploy/yolov7.yaml -------------------------------------------------------------------------------- /cfg/deploy/yolov7x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/deploy/yolov7x.yaml -------------------------------------------------------------------------------- /cfg/training/yolov7-d6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/training/yolov7-d6.yaml -------------------------------------------------------------------------------- /cfg/training/yolov7-e6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/training/yolov7-e6.yaml -------------------------------------------------------------------------------- /cfg/training/yolov7-e6e.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/training/yolov7-e6e.yaml -------------------------------------------------------------------------------- /cfg/training/yolov7-tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/training/yolov7-tiny.yaml -------------------------------------------------------------------------------- /cfg/training/yolov7-w6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/training/yolov7-w6.yaml -------------------------------------------------------------------------------- /cfg/training/yolov7.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/training/yolov7.yaml -------------------------------------------------------------------------------- /cfg/training/yolov7x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/cfg/training/yolov7x.yaml -------------------------------------------------------------------------------- /data/coco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/data/coco.yaml -------------------------------------------------------------------------------- /data/hyp.scratch.custom.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/data/hyp.scratch.custom.yaml -------------------------------------------------------------------------------- /data/hyp.scratch.p5.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/data/hyp.scratch.p5.yaml -------------------------------------------------------------------------------- /data/hyp.scratch.p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/data/hyp.scratch.p6.yaml -------------------------------------------------------------------------------- /data/hyp.scratch.tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/data/hyp.scratch.tiny.yaml -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/detect.py -------------------------------------------------------------------------------- /export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/export.py -------------------------------------------------------------------------------- /figure/horses_prediction.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/figure/horses_prediction.jpg -------------------------------------------------------------------------------- /figure/mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/figure/mask.png -------------------------------------------------------------------------------- /figure/performance.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/figure/performance.png -------------------------------------------------------------------------------- /figure/pose.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/figure/pose.png -------------------------------------------------------------------------------- /figure/tennis.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/figure/tennis.jpg -------------------------------------------------------------------------------- /figure/tennis_caption.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/figure/tennis_caption.png -------------------------------------------------------------------------------- /figure/tennis_panoptic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/figure/tennis_panoptic.png -------------------------------------------------------------------------------- /figure/tennis_semantic.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/figure/tennis_semantic.jpg -------------------------------------------------------------------------------- /imgANDvideo/friends.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/imgANDvideo/friends.jpg -------------------------------------------------------------------------------- /imgANDvideo/girl.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/imgANDvideo/girl.jpg -------------------------------------------------------------------------------- /imgANDvideo/huaxue.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/imgANDvideo/huaxue.jpg -------------------------------------------------------------------------------- /imgANDvideo/myVideo.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/imgANDvideo/myVideo.mp4 -------------------------------------------------------------------------------- /imgANDvideo/train.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/imgANDvideo/train.png -------------------------------------------------------------------------------- /imgANDvideo/xican.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/imgANDvideo/xican.jpg -------------------------------------------------------------------------------- /imgANDvideo/yuanli.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/imgANDvideo/yuanli.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | # init -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/models/common.py -------------------------------------------------------------------------------- /models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/models/experimental.py -------------------------------------------------------------------------------- /models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/models/yolo.py -------------------------------------------------------------------------------- /scripts/get_coco.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/scripts/get_coco.sh -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/test.py -------------------------------------------------------------------------------- /tools/YOLOv7-Dynamic-Batch-ONNXRUNTIME.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/tools/YOLOv7-Dynamic-Batch-ONNXRUNTIME.ipynb -------------------------------------------------------------------------------- /tools/YOLOv7-Dynamic-Batch-TENSORRT.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/tools/YOLOv7-Dynamic-Batch-TENSORRT.ipynb -------------------------------------------------------------------------------- /tools/YOLOv7CoreML.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/tools/YOLOv7CoreML.ipynb -------------------------------------------------------------------------------- /tools/YOLOv7onnx.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/tools/YOLOv7onnx.ipynb -------------------------------------------------------------------------------- /tools/YOLOv7trt.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/tools/YOLOv7trt.ipynb -------------------------------------------------------------------------------- /tools/compare_YOLOv7_vs_YOLOv5m6.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/tools/compare_YOLOv7_vs_YOLOv5m6.ipynb -------------------------------------------------------------------------------- /tools/compare_YOLOv7_vs_YOLOv5m6_half.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/tools/compare_YOLOv7_vs_YOLOv5m6_half.ipynb -------------------------------------------------------------------------------- /tools/compare_YOLOv7_vs_YOLOv5s6.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/tools/compare_YOLOv7_vs_YOLOv5s6.ipynb -------------------------------------------------------------------------------- /tools/compare_YOLOv7e6_vs_YOLOv5x6.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/tools/compare_YOLOv7e6_vs_YOLOv5x6.ipynb -------------------------------------------------------------------------------- /tools/compare_YOLOv7e6_vs_YOLOv5x6_half.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/tools/compare_YOLOv7e6_vs_YOLOv5x6_half.ipynb -------------------------------------------------------------------------------- /tools/instance.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/tools/instance.ipynb -------------------------------------------------------------------------------- /tools/keypoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/tools/keypoint.ipynb -------------------------------------------------------------------------------- /tools/reparameterization.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/tools/reparameterization.ipynb -------------------------------------------------------------------------------- /tools/visualization.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/tools/visualization.ipynb -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | # init -------------------------------------------------------------------------------- /utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/activations.py -------------------------------------------------------------------------------- /utils/add_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/add_nms.py -------------------------------------------------------------------------------- /utils/autoanchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/autoanchor.py -------------------------------------------------------------------------------- /utils/aws/__init__.py: -------------------------------------------------------------------------------- 1 | #init -------------------------------------------------------------------------------- /utils/aws/mime.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/aws/mime.sh -------------------------------------------------------------------------------- /utils/aws/resume.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/aws/resume.py -------------------------------------------------------------------------------- /utils/aws/userdata.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/aws/userdata.sh -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/google_app_engine/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/google_app_engine/Dockerfile -------------------------------------------------------------------------------- /utils/google_app_engine/additional_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/google_app_engine/additional_requirements.txt -------------------------------------------------------------------------------- /utils/google_app_engine/app.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/google_app_engine/app.yaml -------------------------------------------------------------------------------- /utils/google_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/google_utils.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/plots.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /utils/wandb_logging/__init__.py: -------------------------------------------------------------------------------- 1 | # init -------------------------------------------------------------------------------- /utils/wandb_logging/log_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/wandb_logging/log_dataset.py -------------------------------------------------------------------------------- /utils/wandb_logging/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zxx1218/ObjectDetect/HEAD/utils/wandb_logging/wandb_utils.py --------------------------------------------------------------------------------