DDH-YOLOv5: Improved YOLOv5 Based on Double IoU-aware Decoupled Head for Object Detection

Wang Hui,Jin Yang,Ke Hongchang,Zhang Xinping
DOI: https://doi.org/10.1007/s11554-022-01241-z
IF: 2.293
2022-01-01
Journal of Real-Time Image Processing
Abstract:YOLOv5 is a high-performance real-time object detector that plays an important role in one-stage detectors. However, there are two problems with the design of the YOLOv5 head. The common branch of classification task and regression task of the YOLOv5 head will hurt the training process, and the correlation between classification score and localization accuracy is low. We propose a Double IoU-aware Decoupled Head (DDH) and apply it to YOLOv5. The improved model is named DDH-YOLOv5, which substantially improves the localization accuracy of the model without significantly increasing FLOPS and parameters. Extensive experiments on dataset PASCAL VOC2007 show that DDH-YOLOv5 has good performance. Compared with YOLOv5, DDH-YOLOv5m and DDH-YOLOv5l proposed in this paper achieve 2.4 % and 1.3 % improvement in Average Precision (AP), respectively. Compared with Deformable DETR, which is known for its fast-converging, DDH-YOLOv5 completely outperforms Deformable DETR on COCO2017 Val with half of FLOPS and only a quarter of epochs.
What problem does this paper attempt to address?