N-IoU: better IoU-based bounding box regression loss for object detection

被引:6
|
作者
Su, Keke [1 ,2 ]
Cao, Lihua [1 ]
Zhao, Botong [3 ]
Li, Ning [1 ]
Wu, Di [1 ]
Han, Xiyu [1 ]
机构
[1] Chinese Acad Sci, Changchun Inst Opt Fine Mech & Phys, Southeast Lake Rd, Changchun 130033, Peoples R China
[2] Univ Chinese Acad Sci, Beijing 100049, Peoples R China
[3] East China Normal Univ, Sch Commun & Elect Engn, Shanghai 200241, Peoples R China
来源
NEURAL COMPUTING & APPLICATIONS | 2024年 / 36卷 / 06期
关键词
Bounding box regression; IoU; N-IoU; Regression loss; Object detection; Computer vision; NETWORKS;
D O I
10.1007/s00521-023-09133-4
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Object detection is one of the core tasks of computer vision, and bounding box (bbox) regression is one of the basic tasks of object detection. In recent years of related research, bbox regression is often used in the Intersection over Union (IoU) loss and its improved version. In this paper, for the first time, we introduce the Dice coefficient into the regression loss calculation and propose a new measure which is superior to and can replace the IoU. We define three properties of the new measure and prove the theory by mathematical reasoning and analysis of the existing work. This paper also proposes the N-IoU regression loss family. And the superiority of the N-IoU regression loss family is proved by designing simulation experiments and comparative experiments. The main results of this paper are: (1) The proposed new measure is better than IoU which can be used to evaluate bounding box regression, and the three properties of the new measure can be used as a broad criterion for the design of regression loss functions; and (2) we propose N-IoU loss. The parameter n of N-IOU can be debugged, which can be widely adapted to different application scenarios with higher flexibility, and the regression performance is better.
引用
收藏
页码:3049 / 3063
页数:15
相关论文
共 50 条
  • [41] BIoU: An Improved Bounding Box Regression for Object Detection
    Ravi, Niranjan
    Naqvi, Sami
    El-Sharkawy, Mohamed
    JOURNAL OF LOW POWER ELECTRONICS AND APPLICATIONS, 2022, 12 (04)
  • [42] ROBUST BOUNDING BOX REGRESSION FOR SMALL OBJECT DETECTION
    Guo, Ziqi
    He, Chu
    Zhou, Lian
    Zhang, Qingyi
    Sun, Shilei
    2023 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING, ICIP, 2023, : 2290 - 2294
  • [43] Bounding Box Regression with Uncertainty for Accurate Object Detection
    He, Yihui
    Zhu, Chenchen
    Wang, Jianren
    Savvides, Marios
    Zhang, Xiangyu
    2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, : 2883 - 2892
  • [44] A double branches network with IoU-based loss weight to get more accurate deformable pooling feature for angle sensitive ships
    Wang, Tong
    Li, Ying
    GEOCARTO INTERNATIONAL, 2022, 37 (26) : 14528 - 14546
  • [45] Unmanned aerial vehicle (UAV) object detection algorithm based on keypoints representation and rotated distance-IoU loss
    Hufei Zhu
    Yonghui Huang
    Ying Xu
    Jianhong Zhou
    Fuqin Deng
    Yikui Zhai
    Journal of Real-Time Image Processing, 2024, 21
  • [46] Unmanned aerial vehicle (UAV) object detection algorithm based on keypoints representation and rotated distance-IoU loss
    Zhu, Hufei
    Huang, Yonghui
    Xu, Ying
    Zhou, Jianhong
    Deng, Fuqin
    Zhai, Yikui
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2024, 21 (02)
  • [47] Object Localization Using Vision Transformer with a Loss Function Based on IOU and Mean Squared Error
    Cheng, Hong
    Cheng, Wen
    2023 INTERNATIONAL CONFERENCE ON COMPUTATIONAL SCIENCE AND COMPUTATIONAL INTELLIGENCE, CSCI 2023, 2023, : 1284 - 1288
  • [48] Multi-branch Bounding Box Regression for Object Detection
    Hui-Shen Yuan
    Si-Bao Chen
    Bin Luo
    Hao Huang
    Qiang Li
    Cognitive Computation, 2023, 15 : 1300 - 1307
  • [49] Multi-branch Bounding Box Regression for Object Detection
    Yuan, Hui-Shen
    Chen, Si-Bao
    Luo, Bin
    Huang, Hao
    Li, Qiang
    COGNITIVE COMPUTATION, 2023, 15 (04) : 1300 - 1307
  • [50] Bounding Box Projection for Regression Uncertainty in Oriented Object Detection
    Wu, Qian
    Xiang, Wangtao
    Tang, Rui
    Zhu, Jun
    IEEE ACCESS, 2021, 9 : 58768 - 58779