Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

YOLOv4: Optimal Speed and Accuracy of Object Detection

About

There are a huge number of features which are said to improve Convolutional Neural Network (CNN) accuracy. Practical testing of combinations of such features on large datasets, and theoretical justification of the result, is required. Some features operate on certain models exclusively and for certain problems exclusively, or only for small-scale datasets; while some features, such as batch-normalization and residual-connections, are applicable to the majority of models, tasks, and datasets. We assume that such universal features include Weighted-Residual-Connections (WRC), Cross-Stage-Partial-connections (CSP), Cross mini-Batch Normalization (CmBN), Self-adversarial-training (SAT) and Mish-activation. We use new features: WRC, CSP, CmBN, SAT, Mish activation, Mosaic data augmentation, CmBN, DropBlock regularization, and CIoU loss, and combine some of them to achieve state-of-the-art results: 43.5% AP (65.7% AP50) for the MS COCO dataset at a realtime speed of ~65 FPS on Tesla V100. Source code is at https://github.com/AlexeyAB/darknet

Alexey Bochkovskiy, Chien-Yao Wang, Hong-Yuan Mark Liao• 2020

Related benchmarks

TaskDatasetResultRank
Object DetectionCOCO 2017 (val)
AP21.7
2454
Object DetectionCOCO (test-dev)
mAP55.8
1195
Object DetectionMS COCO (test-dev)
mAP@.565.7
677
Object DetectionCOCO (val)
mAP43.5
613
Object DetectionCOCO v2017 (test-dev)
mAP43.5
499
Instance SegmentationCOCO 2017 (test-dev)--
253
Object DetectionVisDrone
mAP5030.7
26
Object DetectionSESYD Floorplans (test)
AP5093.04
21
Object DetectionPKU-DDD Car 17
mAP5081.3
20
DetectionKUMC
F1 Score57.2
20
Showing 10 of 18 rows

Other info

Code

Follow for update