Keypoint regression strategy and angle loss based YOLO for object detection
Xiuling Wang, Lingkun Kong, Zhiguo Zhang, Haixia Wang, Xiao Lu
Abstract
The YOLOv4 approach has gained significant popularity in industrial object detection due to its impressive real-time processing speed and relatively favorable accuracy. However, it has been observed that YOLOv4 faces challenges in accurately detecting small objects. Its bounding box regression strategy is rigid and fails to effectively leverage the asymmetric characteristics of objects, limiting its ability to enhance object detection accuracy. This paper proposes an enhanced version of YOLOv4 called KR-AL-YOLO (keypoint regression strategy and angle loss based YOLOv4). The KR-AL-YOLO approach introduces two customized modules: an keypoint regression strategy and an angle-loss function. These modules contribute to improving the algorithm's detection accuracy by enabling more precise localization of objects. Additionally, KR-AL-YOLO adopts an improved feature fusion technique, which facilitates enhanced information flow within the network, thereby further enhancing accuracy performance. Experimental evaluations conducted on the COCO2017 dataset demonstrate the effectiveness of the proposed method. KR-AL-YOLO achieves an average precision of 45.6%, surpassing both YOLOv4 and certain previously developed one-stage detectors. The utilization of keypoint regression strategy and the incorporation of robust feature fusion contribute to superior object detection accuracy in KR-AL-YOLO compared to YOLOv4.