YOLOv4
YOLOv4 consists of: https://arxiv.org/abs/2004.10934
- Backbone: CSPDarknet53
- Neck: SPP, PAN
- Head: YOLOv3
Download:
-
cfg-file: https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.cfg
-
weights-file: https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.weights
-
pre-trained weights-file for training: https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.conv.137
Accuracy (AP) / Speed (FPS):
YOLOv4 (416x416 batch=1 FP16) - 32 FPS on Jetson Xavier AGX by using tkDNN+TensorRT: https://github.com/ceccocats/tkDNN
| Size | Darknet FPS (avg) | tkDNN TensorRT FP32 FPS | tkDNN TensorRT FP16 FPS | tkDNN TensorRT FP16 batch=4 FPS | Speedup |
|---|---|---|---|---|---|
| 320 | 100.6 | 116 | 202 | 423 | 4.2x |
| 416 | 82.5 | 103 | 162 | 284 | 3.5x |
| 512 | 69.7 | 91 | 134 | 206 | 2.9x |
| 680 | 53.6 | 62 | 100 | 150 | 2.8x |
