Location IoU: A New Evaluation and Loss for Bounding Box Regression in Object Detection
Yang Lu, Kai Zhang, Jiaqi Liu, Chongke Bi · 2024
In the field of object detection bounding box regression, IoU (Intersection over Union) is a commonly used evaluation metric for measuring the overlap between predicted and ground truth bounding boxes. One limitation of IoU, however, is that the gradient goes to zero when there is no intersection between bounding boxes. Recent evaluation metrics have been developed to overcome this problem by incorporating penalty terms and other techniques for improving IoU. Despite these advances, existing methods still suffer to some degree from the problems of small gradients and slow rate of convergence. In this paper, a new evaluation method named LIoU (Location IoU) is proposed to address these issues. It can maximize the shared area of the intersection when used as the loss function and incorporate a variable gradient parameter to fit to datasets of different sizes. The experimental results indicate that we achieve 0.27% and 3.48% accuracy improvements respectively in SSD and YOLOv5 network on the VOC dataset, and 2.23% in the YOLOv5 network on the COCO dataset. The results of this study highlight the effectiveness of LIoU in improving the performance of object detection models.