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 条
  • [21] Selective feature block and joint IoU loss for object detection
    Wang, Junyi
    Hua, Ruzhao
    Jiang, Xuezheng
    Song, Kechen
    Meng, Qinggang
    Saada, Mohamad
    TRANSACTIONS OF THE INSTITUTE OF MEASUREMENT AND CONTROL, 2024, 46 (14) : 2757 - 2767
  • [22] M2IoU: A Min-Max Distance-based Loss Function for Bounding Box Regression in Medical Imaging
    Shandilya, Anurag Kumar
    Shah, Kalash
    Kanekar, Bhavik
    Gautam, Akshat
    Tandon, Pavni
    Ramakrishnan, Ganesh
    Jadhav, Kshitij
    PROCEEDINGS OF THE 33RD ACM INTERNATIONAL CONFERENCE ON INFORMATION AND KNOWLEDGE MANAGEMENT, CIKM 2024, 2024, : 4041 - 4045
  • [23] IoU_MDA: An Occluded Object Detection Algorithm Based on Fuzzy Sample Anchor Box IoU Matching Degree Deviation Aware
    Chen, Yuling
    Li, Xiaoxia
    He, Zhenxiang
    Chen, Hang
    Chen, Jinwei
    Wu, Bin
    IEEE ACCESS, 2024, 12 : 47630 - 47645
  • [24] IA-CIOU: An Improved IOU Bounding Box Loss Function for SAR Ship Target Detection Methods
    Huang, Pingping
    Tian, Shihao
    Su, Yun
    Tan, Weixian
    Dong, Yifan
    Xu, Wei
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2024, 17 : 10569 - 10582
  • [25] Scale-Sensitive IOU Loss: An Improved Regression Loss Function in Remote Sensing Object Detection
    Du, Shuangjiang
    Zhang, Baofu
    Zhang, Pin
    IEEE ACCESS, 2021, 9 : 141258 - 141272
  • [26] Gaussian guided IoU: A better metric for balanced learning on object detection
    Gou, Lijun
    Wu, Shengkai
    Yang, Jinrong
    Yu, Hangcheng
    Li, Xiaoping
    IET COMPUTER VISION, 2022, 16 (06) : 556 - 566
  • [27] Alpha-IoU: A Family of Power Intersection over Union Losses for Bounding Box Regression
    He, Jiabo
    Erfani, Sarah
    Ma, Xingjun
    Bailey, James
    Chi, Ying
    Hua, Xian-Sheng
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 34 (NEURIPS 2021), 2021, 34
  • [28] Mixed geometric loss for bounding box regression in object detection
    Ren, Xudie
    Luo, Fucai
    Li, Shenghong
    JOURNAL OF ELECTRONIC IMAGING, 2020, 29 (05)
  • [29] Effective ML-Block and Weighted IoU Loss for Object Detection
    Yuan, Zhenxin
    Xiao, Xin
    Zhao, Shuai
    Jiang, Linlin
    Li, Zhongtao
    ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING - ICANN 2022, PT IV, 2022, 13532 : 224 - 235
  • [30] Object Detection for UAV Aerial Scenarios Based on Vectorized IOU
    Lu, Shun
    Lu, Hanyu
    Dong, Jun
    Wu, Shuang
    SENSORS, 2023, 23 (06)