YOLOX: Exceeding YOLO Series in 2021

18 Jul 2021  ·  Zheng Ge, Songtao Liu, Feng Wang, Zeming Li, Jian Sun ·

In this report, we present some experienced improvements to YOLO series, forming a new high-performance detector -- YOLOX. We switch the YOLO detector to an anchor-free manner and conduct other advanced detection techniques, i.e., a decoupled head and the leading label assignment strategy SimOTA to achieve state-of-the-art results across a large scale range of models: For YOLO-Nano with only 0.91M parameters and 1.08G FLOPs, we get 25.3% AP on COCO, surpassing NanoDet by 1.8% AP; for YOLOv3, one of the most widely used detectors in industry, we boost it to 47.3% AP on COCO, outperforming the current best practice by 3.0% AP; for YOLOX-L with roughly the same amount of parameters as YOLOv4-CSP, YOLOv5-L, we achieve 50.0% AP on COCO at a speed of 68.9 FPS on Tesla V100, exceeding YOLOv5-L by 1.8% AP. Further, we won the 1st Place on Streaming Perception Challenge (Workshop on Autonomous Driving at CVPR 2021) using a single YOLOX-L model. We hope this report can provide useful experience for developers and researchers in practical scenes, and we also provide deploy versions with ONNX, TensorRT, NCNN, and Openvino supported. Source code is at https://github.com/Megvii-BaseDetection/YOLOX.

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Uses Extra
Training Data
Result Benchmark
Real-Time Object Detection Argoverse-HD (Detection-Only, Test) YOLOX AP 41.1 # 1
Real-Time Object Detection Argoverse-HD (Detection-Only, Val) YOLOX AP 47.42 # 1
Real-Time Object Detection Argoverse-HD (Full-Stack, Test) YOLOX AP 41.1 # 1
2D Object Detection CeyMo YOLOX mAP 57.7 # 4
Object Detection COCO-O YOLOX-S Average mAP 20.6 # 29
Effective Robustness 2.48 # 28
Object Detection COCO-O YOLOX-X Average mAP 30.3 # 19
Effective Robustness 7.26 # 16
Object Detection COCO test-dev YOLOX-X (Modified CSP v5) box mAP 51.2 # 78
AP50 69.6 # 39
AP75 55.7 # 37
APS 31.2 # 43
APM 56.1 # 21
APL 66.1 # 19
Params (M) 99.1 # 7
Object Detection COCO test-dev YOLOX-Darknet53(Darknet53, 640x640, single-scale) box mAP 48.0 # 106
Object Detection COCO test-dev YOLOX-Nano(416x416, single-scale) box mAP 25.8 # 232
Params (M) 0.91 # 11
Object Detection COCO test-dev YOLOX-x(Modified CSP v5, 640x640, single-scale) box mAP 51.5 # 73
Object Detection COCO test-dev YOLOX-Tiny(416x416, single-scale) box mAP 32.8 # 227
Params (M) 5.06 # 10
Real-Time Object Detection MS COCO YOLOv5-X FPS (V100, b=1) 62.5 # 16
box AP 50.4 # 28
Object Detection WaterScenes YOLOX-M mAP@50-95 57.8 # 2

Methods