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

YOLOv10: Real-Time End-to-End Object Detection

About

Over the past years, YOLOs have emerged as the predominant paradigm in the field of real-time object detection owing to their effective balance between computational cost and detection performance. Researchers have explored the architectural designs, optimization objectives, data augmentation strategies, and others for YOLOs, achieving notable progress. However, the reliance on the non-maximum suppression (NMS) for post-processing hampers the end-to-end deployment of YOLOs and adversely impacts the inference latency. Besides, the design of various components in YOLOs lacks the comprehensive and thorough inspection, resulting in noticeable computational redundancy and limiting the model's capability. It renders the suboptimal efficiency, along with considerable potential for performance improvements. In this work, we aim to further advance the performance-efficiency boundary of YOLOs from both the post-processing and model architecture. To this end, we first present the consistent dual assignments for NMS-free training of YOLOs, which brings competitive performance and low inference latency simultaneously. Moreover, we introduce the holistic efficiency-accuracy driven model design strategy for YOLOs. We comprehensively optimize various components of YOLOs from both efficiency and accuracy perspectives, which greatly reduces the computational overhead and enhances the capability. The outcome of our effort is a new generation of YOLO series for real-time end-to-end object detection, dubbed YOLOv10. Extensive experiments show that YOLOv10 achieves state-of-the-art performance and efficiency across various model scales. For example, our YOLOv10-S is 1.8$\times$ faster than RT-DETR-R18 under the similar AP on COCO, meanwhile enjoying 2.8$\times$ smaller number of parameters and FLOPs. Compared with YOLOv9-C, YOLOv10-B has 46\% less latency and 25\% fewer parameters for the same performance.

Ao Wang, Hui Chen, Lihao Liu, Kai Chen, Zijia Lin, Jungong Han, Guiguang Ding• 2024

Related benchmarks

TaskDatasetResultRank
Object DetectionCOCO 2017 (val)
AP54.9
2454
Object DetectionCOCO (val)--
613
Object DetectionPASCAL VOC 2007+2012 (test)
mAP (mean Average Precision)60.6
95
Object DetectionVisDrone (val)
AP5044.5
66
Monocular 3D Object DetectionKITTI car category (val)
AP 3D (R40)19.98
37
Object DetectionMS-COCO 2017 (val)
Box AP54.4
32
3D Object DetectionRope3D (val)
AP (IoU=0.5, Car)44.06
31
Object DetectionUAVDB (test)
AP5084.2
31
Object DetectionUAVDB (val)
AP5081.7
31
Object DetectionSAR-Aircraft v1.0 (test)--
27
Showing 10 of 48 rows

Other info

Code

Follow for update