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 条
  • [41] Structured Modeling of Joint Deep Feature and Prediction Refinement for Salient Object Detection
    Xu, Yingyue
    Xu, Dan
    Hong, Xiaopeng
    Ouyang, Wanli
    Ji, Rongrong
    Xu, Min
    Zhao, Guoying
    2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 3788 - 3797
  • [42] Joint Image and Feature Enhancement for Object Detection under Adverse Weather Conditions
    Yin, Mengyu
    Ling, Mingyang
    Chang, Kan
    Yuan, Zijian
    Qin, Qingpao
    Chen, Boning
    2024 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS, IJCNN 2024, 2024,
  • [43] An Object Detection Method Based on the Joint Feature of the H-S Color Descriptor and the SIFT Feature
    Peng, Yong Kang
    Zhang, Yi Lai
    Cheng, Xi En
    Li, Yi Cheng
    Zhao, Shi Dong
    2018 INTERNATIONAL CONFERENCE ON AUDIO, LANGUAGE AND IMAGE PROCESSING (ICALIP), 2018, : 375 - 379
  • [44] Object Localization Using Vision Transformer with a Loss Function Based on IOU and Mean Squared Error
    Cheng, Hong
    Cheng, Wen
    2023 INTERNATIONAL CONFERENCE ON COMPUTATIONAL SCIENCE AND COMPUTATIONAL INTELLIGENCE, CSCI 2023, 2023, : 1284 - 1288
  • [45] Cross-Domain Joint Object Detection Based on Visual Selective Attention
    Shi, Yi
    Gao, Gangyao
    Qin, Long
    Yan, Hongmei
    INTERNATIONAL JOURNAL OF PSYCHOPHYSIOLOGY, 2021, 168 : S134 - S134
  • [46] Influence of Insufficient Dataset Augmentation on IoU and Detection Threshold in CNN Training for Object Detection on Aerial Images
    Bozko, Arkadiusz
    Ambroziak, Leszek
    SENSORS, 2022, 22 (23)
  • [47] CSDA: Learning Category-Scale Joint Feature for Domain Adaptive Object Detection
    Gao, Changlong
    Liu, Chengxu
    Dun, Yujie
    Qian, Xueming
    2023 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2023), 2023, : 11387 - 11396
  • [48] Pseudo-IoU: Improving Label Assignment in Anchor-Free Object Detection
    Li, Jiachen
    Cheng, Bowen
    Feris, Rogerio
    Xiong, Jinjun
    Huang, Thomas S.
    Hwu, Wen-Mei
    Shi, Humphrey
    2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS, CVPRW 2021, 2021, : 2378 - 2387
  • [49] IoU-Enhanced Attention for End-to-End Task Specific Object Detection
    Zhao, Jing
    Wu, Shengjian
    Sun, Li
    Li, Qingli
    COMPUTER VISION - ACCV 2022, PT V, 2023, 13845 : 124 - 141
  • [50] Non-binary IoU and progressive coupling and refining network for salient object detection
    Zhou, Qianwei
    Zhou, Chen
    Yang, Zihao
    Xu, Yingkun
    Guan, Qiu
    EXPERT SYSTEMS WITH APPLICATIONS, 2023, 230