DeepAI AI Chat
Log In Sign Up

Wise-IoU: Bounding Box Regression Loss with Dynamic Focusing Mechanism

by   Zanjia Tong, et al.

The loss function for bounding box regression (BBR) is essential to object detection. Its good definition will bring significant performance improvement to the model. Most existing works assume that the examples in the training data are high-quality and focus on strengthening the fitting ability of BBR loss. If we blindly strengthen BBR on low-quality examples, it will jeopardize localization performance. Focal-EIoU v1 was proposed to solve this problem, but due to its static focusing mechanism (FM), the potential of non-monotonic FM was not fully exploited. Based on this idea, we propose an IoU-based loss with a dynamic non-monotonic FM named Wise-IoU (WIoU). When WIoU is applied to the state-of-the-art real-time detector YOLOv7, the AP-75 on the MS-COCO dataset is improved from 53.03


page 2

page 5

page 6


Softer-NMS: Rethinking Bounding Box Regression for Accurate Object Detection

Non-maximum suppression (NMS) is essential for state-of-the-art object d...

MPDIoU: A Loss for Efficient and Accurate Bounding Box Regression

Bounding box regression (BBR) has been widely used in object detection a...

Gradient Harmonized Single-stage Detector

Despite the great success of two-stage detectors, single-stage detector ...

Control Distance IoU and Control Distance IoU Loss Function for Better Bounding Box Regression

Numerous improvements for feedback mechanisms have contributed to the gr...

Decoupled IoU Regression for Object Detection

Non-maximum suppression (NMS) is widely used in object detection pipelin...

SIoU Loss: More Powerful Learning for Bounding Box Regression

The effectiveness of Object Detection, one of the central problems in co...

PBRnet: Pyramidal Bounding Box Refinement to Improve Object Localization Accuracy

Many recently developed object detectors focused on coarse-to-fine frame...