PyTorch Object Detection
YOLOv5 is Here
YOLOv5 is smaller and generally easier to use in production. Given it is natively implemented in PyTorch (rather than Darknet), modifying the architecture and exporting to many deploy environments is straightforward.
- SIZE: YOLOv5s is about 88% smaller than big-YOLOv4 (27 MB vs 244 MB)
SPEED: YOLOv5 performs batch inference at about 140 FPS by default.
ACCURACY: YOLOv5 is roughly as accurate as YOLOv4 on small tasks (0.895 mAP vs 0.892 mAP on BCCD). On larger tasks like COCO, YOLOv4 is more performant.
Read more about YOLOv5 performance.