YOLOv4 pre-release
AlexeyAB
released this
15 May 14:17
·
434 commits
to master
since this release
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 |