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 条
  • [1] N-IoU: better IoU-based bounding box regression loss for object detection
    Keke Su
    Lihua Cao
    Botong Zhao
    Ning Li
    Di Wu
    Xiyu Han
    Neural Computing and Applications, 2024, 36 : 3049 - 3063
  • [2] Absolute size IoU loss for the bounding box regression of the object detection
    Tian, Di
    Han, Yi
    Wang, Shu
    Chen, Xu
    Guan, Tian
    NEUROCOMPUTING, 2022, 500 : 1029 - 1040
  • [3] Control Distance IoU and Control Distance IoU Loss for Better Bounding Box Regression
    Dong, Chen
    Miao, Duoqian
    PATTERN RECOGNITION, 2023, 137
  • [4] Ellipse IoU Loss: Better Learning for Rotated Bounding Box Regression
    Li, Wenjie
    Shang, Ronghua
    Ju, Zihan
    Feng, Jie
    Xu, Songhua
    Zhang, Weitong
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2024, 21 : 1 - 5
  • [5] Distance-IoU Loss: Faster and Better Learning for Bounding Box Regression
    Zheng, Zhaohui
    Wang, Ping
    Liu, Wei
    Li, Jinze
    Ye, Rongguang
    Ren, Dongwei
    THIRTY-FOURTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, THE THIRTY-SECOND INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE CONFERENCE AND THE TENTH AAAI SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2020, 34 : 12993 - 13000
  • [6] Manhattan-distance IOU loss for fast and accurate bounding box regression and object detection
    Shen, Yanyun
    Zhang, Feizhao
    Liu, Di
    Pu, Weihua
    Zhang, Qingling
    NEUROCOMPUTING, 2022, 500 : 99 - 114
  • [7] Gaussian-IoU loss: Better learning for bounding box regression on PCB component detection
    Liu, Xin
    Hu, Jinshuai
    Wang, Haixia
    Zhang, Zhiguo
    Lu, Xiao
    Sheng, Chunyang
    Song, Shibin
    Nie, Jun
    EXPERT SYSTEMS WITH APPLICATIONS, 2022, 190
  • [8] Improved Design Based on IoU Loss Functions for Bounding Box Regression
    Liu, Zhibo
    Cheng, Jian
    Wang, Qinzheng
    Xian, Lihua
    2022 IEEE 6TH ADVANCED INFORMATION TECHNOLOGY, ELECTRONIC AND AUTOMATION CONTROL CONFERENCE (IAEAC), 2022, : 452 - 458
  • [9] Focal and efficient IOU loss for accurate bounding box regression
    Zhang, Yi-Fan
    Ren, Weiqiang
    Zhang, Zhang
    Jia, Zhen
    Wang, Liang
    Tan, Tieniu
    NEUROCOMPUTING, 2022, 506 (146-157) : 146 - 157
  • [10] Smooth-IoU Loss for Bounding Box Regression in Visual Tracking
    Li G.
    Zhao W.
    Liu P.
    Tang X.-L.
    Zidonghua Xuebao/Acta Automatica Sinica, 2023, 49 (02): : 288 - 306