Anchor constrained refinement network with Intersection-over-Union-aware and scale-aware loss for oriented object detection

被引:0
|
作者
Wu, Qin [1 ,2 ]
Yao, Zikang [1 ]
Chai, Zhilei [1 ,2 ]
机构
[1] Jiangnan Univ, Dept Comp Sci, Wuxi, Jiangsu, Peoples R China
[2] Jiangnan Univ, Jiangsu Prov Engn Lab Pattern Recognit & Computat, Wuxi, Jiangsu, Peoples R China
基金
中国国家自然科学基金;
关键词
anchor constrained refinement; feature alignment; multi-scale feature fusion; IoU-aware loss; scale-aware loss; rotated object detection;
D O I
10.1117/1.JEI.31.1.013029
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Object detection has achieved good progress in the last few years. However, there are many challenges in the field of remote sensing imagery. Objects in remote sensing images usually have arbitrary orientations and various scales. In addition, some objects are easily overwhelmed by a cluttered background. To take advantage of single-stage object detectors that have fast speed, many cascaded structures based on single-stage detectors have been proposed to improve detection performance. However, feature inconsistency in cascade structure results in poor detection performance. To address these problems, we propose an innovative model in terms of both model improvement and loss function refinement. This model consists of an attention module to highlight useful information in cluttered scenes, a multi-scale feature fusion module, and a cascade refinement module with anchor constrained convolution to address feature inconsistency. Furthermore, Intersection-over-Union (IoU) classification loss is proposed to enhance the correlation between classification and localization, and a scale-aware regression loss is proposed to improve the detection performance on objects with different scales. We conducted extensive experiments on both the DOTA dataset and the HRSC2016 dataset, and the experimental results show that our model has advantages compared with current state-of-the-art methods. (C) 2022 SPIE and IS&T
引用
收藏
页数:16
相关论文
共 50 条
  • [1] Anchor-Free Object Detection with Scale-Aware Networks for Autonomous Driving
    Piao, Zhengquan
    Wang, Junbo
    Tang, Linbo
    Zhao, Baojun
    Zhou, Shichao
    [J]. ELECTRONICS, 2022, 11 (20)
  • [2] Scale-Aware Trident Networks for Object Detection
    Li, Yanghao
    Chen, Yuntao
    Wang, Naiyan
    Zhang, Zhaoxiang
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 6053 - 6062
  • [3] Scale-aware Automatic Augmentation for Object Detection
    Chen, Yukang
    Li, Yanwei
    Kong, Tao
    Qi, Lu
    Chu, Ruihang
    Li, Lei
    Jia, Jiaya
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 9558 - 9567
  • [4] Global to Local: A Scale-Aware Network for Remote Sensing Object Detection
    Gao, Tao
    Niu, Qianqian
    Zhang, Jing
    Chen, Ting
    Mei, Shaohui
    Jubair, Ahmad
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [5] A Scale-Aware Pyramid Network for Multi-Scale Object Detection in SAR Images
    Tang, Linbo
    Tang, Wei
    Qu, Xin
    Han, Yuqi
    Wang, Wenzheng
    Zhao, Baojun
    [J]. REMOTE SENSING, 2022, 14 (04)
  • [6] Scale-Aware Hierarchical Detection Network for Pedestrian Detection
    Zhang, Xiaowei
    Cao, Shuai
    Chen, Chenglizhao
    [J]. IEEE ACCESS, 2020, 8 : 94429 - 94439
  • [7] Scale-Aware Regional Collective Feature Enhancement Network for Scene Object Detection
    Yiyao Li
    Jin Liu
    Zhenyu Gao
    [J]. Neural Processing Letters, 2023, 55 : 6289 - 6310
  • [8] Scale-Aware Regional Collective Feature Enhancement Network for Scene Object Detection
    Li, Yiyao
    Liu, Jin
    Gao, Zhenyu
    [J]. NEURAL PROCESSING LETTERS, 2023, 55 (05) : 6289 - 6310
  • [9] DHNet: Salient Object Detection With Dynamic Scale-Aware Learning and Hard-Sample Refinement
    Zhang, Chenhao
    Gao, Shanshan
    Mao, Deqian
    Zhou, Yuanfeng
    [J]. IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2022, 32 (11) : 7772 - 7782
  • [10] Scale-aware feature pyramid architecture for marine object detection
    Fengqiang Xu
    Huibing Wang
    Jinjia Peng
    Xianping Fu
    [J]. Neural Computing and Applications, 2021, 33 : 3637 - 3653