Scaled-YOLOv4: Scaling Cross Stage Partial Network

  • 2020-11-16 15:42:00
  • Chien-Yao Wang, Alexey Bochkovskiy, Hong-Yuan Mark Liao
  • 38

Abstract

We show that the YOLOv4 object detection neural network based on the CSPapproach, scales both up and down and is applicable to small and large networkswhile maintaining optimal speed and accuracy. We propose a network scalingapproach that modifies not only the depth, width, resolution, but alsostructure of the network. YOLOv4-large model achieves state-of-the-art results:55.4% AP (73.3% AP50) for the MS COCO dataset at a speed of 15 FPS on TeslaV100, while with the test time augmentation, YOLOv4-large achieves 55.8% AP(73.2 AP50). To the best of our knowledge, this is currently the highestaccuracy on the COCO dataset among any published work. The YOLOv4-tiny modelachieves 22.0% AP (42.0% AP50) at a speed of 443 FPS on RTX 2080Ti, while byusing TensorRT, batch size = 4 and FP16-precision the YOLOv4-tiny achieves 1774FPS.

 

Quick Read (beta)

loading the full paper ...