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 条
  • [21] Joint Pose Estimator and Feature Learning for Object Detection
    Ali, Karim
    Fleuret, Francois
    Hasler, David
    Fua, Pascal
    2009 IEEE 12TH INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2009, : 1373 - 1380
  • [22] Improving Object Localization with Fitness NMS and Bounded IoU Loss
    Tychsen-Smith, Lachlan
    Petersson, Lars
    2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 6877 - 6885
  • [23] Feature Pyramid Reconfiguration With Consistent Loss for Object Detection
    Sun, Fuchun
    Kong, Tao
    Huang, Wenbing
    Tan, Chuanqi
    Fang, Bin
    Liu, Huaping
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2019, 28 (10) : 5041 - 5051
  • [24] Selective feature fusion network for salient object detection
    Sun, Fengming
    Yuan, Xia
    Zhao, Chunxia
    IET COMPUTER VISION, 2023, 17 (04) : 483 - 495
  • [25] Object Detection for UAV Aerial Scenarios Based on Vectorized IOU
    Lu, Shun
    Lu, Hanyu
    Dong, Jun
    Wu, Shuang
    SENSORS, 2023, 23 (06)
  • [26] A feature enriching object detection framework with weak segmentation loss
    Zhang, Tianqi
    Hao, Li-Ying
    Guo, Ge
    NEUROCOMPUTING, 2019, 335 : 72 - 80
  • [27] FFBNET : LIGHTWEIGHT BACKBONE FOR OBJECT DETECTION BASED FEATURE FUSION BLOCK
    Fan, Binqi
    Chen, Yuhao
    Qu, Jianfeng
    Chai, Yi
    Xiao, Chen
    Huang, Pengfei
    2019 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP), 2019, : 3920 - 3924
  • [28] JFDI: Joint Feature Differentiation and Interaction for domain adaptive object detection
    Qiao, Ziteng
    Shi, Dianxi
    Jin, Songchang
    Shi, Yanyan
    Wang, Zhen
    Qiu, Chunping
    NEURAL NETWORKS, 2024, 180
  • [29] Joint Spatial and Temporal Feature Enhancement Network for Disturbed Object Detection
    Zhang, Fan
    Ji, Hongbing
    Zhang, Yongquan
    Zhu, Zhigang
    IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2024, 34 (12) : 12258 - 12273
  • [30] 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