Selective feature block and joint IoU loss for object detection

被引:0
|
作者
Wang, Junyi [1 ,2 ]
Hua, Ruzhao [1 ]
Jiang, Xuezheng [1 ]
Song, Kechen [3 ]
Meng, Qinggang [4 ]
Saada, Mohamad [4 ]
机构
[1] Northeastern Univ, Fac Robot Sci & Engn, 195 Chuangxin Rd, Shenyang 110819, Peoples R China
[2] Northeastern Univ, Foshan Grad Sch Innovat, Shenyang, Peoples R China
[3] Northeastern Univ, Sch Mech Engn & Automat, Shenyang, Peoples R China
[4] Loughborough Univ, Dept Comp Sci, Loughborough, England
基金
中国国家自然科学基金;
关键词
JIoU loss; bounding box regression; SFBlock; feature fusion; object detection; REAL-TIME DETECTION;
D O I
10.1177/01423312241261087
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Object detection is an important problem in the field of computer vision, and feature fusion and bounding box regression are indispensable in mainstream object detection approaches. However, some detectors adopt Feature Pyramid Network, which increases training and detection time. In terms of the regression loss function, some recent techniques based on Intersection over Union (IoU) loss have negative effects on bounding box regression. To overcome these shortcomings, we propose Selective Feature Block (SFBlock) and Joint IoU (JIoU) loss in this article. The proposed SFBlock adaptively selects the features extracted from the Backbone and fuses them into a new feature. We add a penalty term of the intersection area between the prediction box and the target box on Generalized IoU (GIoU) loss to solve the problem that GIoU loss degenerates into IoU loss when the prediction box and the target box are surrounded by each other. A large number of ablation experiments and comparative experiments are carried out to prove the effectiveness of the proposed methods on various models and datasets.
引用
收藏
页码:2757 / 2767
页数:11
相关论文
共 50 条
  • [1] 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
  • [2] Diag-IoU Loss for Object Detection
    Zhang, Shuangqing
    Li, Chenglong
    Jia, Zhen
    Liu, Lei
    Zhang, Zhang
    Wang, Liang
    IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2023, 33 (12) : 7671 - 7683
  • [3] 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
  • [4] 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
  • [5] N-IoU: better IoU-based bounding box regression loss for object detection
    Su, Keke
    Cao, Lihua
    Zhao, Botong
    Li, Ning
    Wu, Di
    Han, Xiyu
    NEURAL COMPUTING & APPLICATIONS, 2024, 36 (06): : 3049 - 3063
  • [6] 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
  • [7] IoU-Balanced loss functions for single-stage object detection
    Wu, Shengkai
    Yang, Jinrong
    Wang, Xinggang
    Li, Xiaoping
    PATTERN RECOGNITION LETTERS, 2022, 156 : 96 - 103
  • [8] 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
  • [9] Addressing the Gaps of IoU Loss in 3D Object Detection with IIoU
    Ravi, Niranjan
    El-Sharkawy, Mohamed
    FUTURE INTERNET, 2023, 15 (12)
  • [10] Decoupled IoU Regression for Object Detection
    Gao, Yan
    Wang, Qimeng
    Tang, Xu
    Wang, Haochen
    Ding, Fei
    Li, Jing
    Hu, Yao
    PROCEEDINGS OF THE 29TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA, MM 2021, 2021, : 5628 - 5636