ultralytics/yolov5: Initial Release

YOLOv5 1.0 Release Notes

• June 22, 2020: PANet updates: increased layers, reduced parameters, faster inference and improved mAP 364fcfd.
• June 19, 2020: FP16 as new default for smaller checkpoints and faster inference d4c6674.
• June 9, 2020: CSP updates: improved speed, size, and accuracy. Credit to @WongKinYiu for excellent CSP work.
• May 27, 2020: Public release of repo. YOLOv5 models are SOTA among all known YOLO implementations.
• April 1, 2020: Start development of future YOLOv3/YOLOv4-based PyTorch models in a range of compound-scaled sizes.

<img src="https://user-images.githubusercontent.com/26833433/85340570-30360a80-b49b-11ea-87cf-bdf33d53ae15.png" width="800"> ** GPU Speed measures end-to-end time per image averaged over 5000 COCO val2017 images using a V100 GPU with batch size 8, and includes image preprocessing, PyTorch FP16 inference, postprocessing and NMS.

Pretrained Checkpoints Model AP^val AP^test AP₅₀ Speed_GPU FPS_GPU params FLOPS YOLOv5s 36.6 36.6 55.8 2.1ms 476 7.5M 13.2B YOLOv5m 43.4 43.4 62.4 3.0ms 333 21.8M 39.4B YOLOv5l 46.6 46.7 65.4 3.9ms 256 47.8M 88.1B YOLOv5x 48.4 48.4 66.9 6.1ms 164 89.0M 166.4B YOLOv3-SPP 45.6 45.5 65.2 4.5ms 222 63.0M 118.0B

AP^test denotes COCO test-dev2017 server results, all other AP results in the table denote val2017 accuracy. All AP numbers are for single-model single-scale without ensemble or test-time augmentation. Reproduce by python test.py --img 736 --conf 0.001
Speed_GPU measures end-to-end time per image averaged over 5000 COCO val2017 images using a GCP n1-standard-16 instance with one V100 GPU, and includes image preprocessing, PyTorch FP16 image inference at --batch-size 32 --img-size 640, postprocessing and NMS. Average NMS time included in this chart is 1-2ms/img. Reproduce by python test.py --img 640 --conf 0.1 All checkpoints are trained to 300 epochs with default settings and hyperparameters (no autoaugmentation).