#254 · Primary category: Computer Vision

ScaledYOLOv4

deep-learning machine-learning ml object-detection pytorch scaled-yolov4 yolo yolov3 yolov4 yolov4-csp yolov4-large yolov4-tiny

Scaled-YOLOv4: Scaling Cross Stage Partial Network

Project last updated:11/03/24

GitHub Stars

2.0K

Forks

563

Contributors

1

License

GPL-3.0

Why we included this project

Object detection teams will find this to be the canonical PyTorch implementation of the scaled YOLOv4 paper, with P5, P6, and P7 models plus the CSP and tiny branches. It ships pretrained COCO weights and straightforward inference scripts, so you can load a model, run it on an image, and get boxes without building the network from scratch. The speed/accuracy tradeoff is easy to reason about: P5 runs around 41 fps at 896 pixels, while P7 goes for higher accuracy at 1536, which matters when picking between edge hardware and a server pipeline. The README's AP tables and per-class metrics let you compare variants before you commit to one.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category