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 条
  • [31] ARIoU: Anchor-free Rotation-decoupling IoU-based optimization for 3D object detection
    Wen, Chenyiming
    Sheng, Hualian
    Zhao, Ming-Min
    Zhao, Min-Jian
    NEUROCOMPUTING, 2024, 594
  • [32] Diagonal-Angle-Foreground IoU Loss Function for Small Object Detection
    Li, Jingpeng
    Li, Hui
    Yang, Mei
    Cai, Delong
    Zhang, Zhaoyun
    ADVANCED INTELLIGENT COMPUTING TECHNOLOGY AND APPLICATIONS, PT XI, ICIC 2024, 2024, 14872 : 108 - 120
  • [33] IoU-Balanced loss functions for single-stage object detection
    Wu, Shengkai
    Yang, Jinrong
    Wang, Xinggang
    Li, Xiaoping
    PATTERN RECOGNITION LETTERS, 2022, 156 : 96 - 103
  • [34] IoU Loss for 2D/3D Object Detection
    Zhou, Dingfu
    Fang, Jin
    Song, Xibin
    Guan, Chenye
    Yin, Junbo
    Dai, Yuchao
    Yang, Ruigang
    2019 INTERNATIONAL CONFERENCE ON 3D VISION (3DV 2019), 2019, : 85 - 94
  • [35] Addressing the Gaps of IoU Loss in 3D Object Detection with IIoU
    Ravi, Niranjan
    El-Sharkawy, Mohamed
    FUTURE INTERNET, 2023, 15 (12)
  • [36] Adaptive occlusion object detection algorithm based on OL-IoU
    Guo, Baicang
    Zhang, Hongyu
    Wang, Huanhuan
    Li, Xinwei
    Jin, Lisheng
    SCIENTIFIC REPORTS, 2024, 14 (01):
  • [37] FPDIoU Loss: A loss function for efficient bounding box regression of rotated object detection
    Ma, Siliang
    Xu, Yong
    IMAGE AND VISION COMPUTING, 2025, 154
  • [38] Enhancing YOLOv5 Performance for Small-Scale Corrosion Detection in Coastal Environments Using IoU-Based Loss Functions
    Yu, Qifeng
    Han, Yudong
    Han, Yi
    Gao, Xinjia
    Zheng, Lingyu
    JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2024, 12 (12)
  • [39] Collaborative learning in bounding box regression for object detection
    Fang, Xian
    Kuang, Zengsheng
    Zhang, Ruixun
    Shao, Xiuli
    Wang, Hongpeng
    PATTERN RECOGNITION LETTERS, 2021, 148 : 121 - 127
  • [40] Bounding box regression with balance for harmonious object detection
    Wang, Chenzhong
    Gong, Xun
    JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2022, 89