A Novel Anchor-Free Detector Using Global Context-Guide Feature Balance Pyramid and United Attention for SAR Ship Detection

被引:29
作者
Bai, Lin [1 ]
Yao, Cheng [1 ]
Ye, Zhen [1 ]
Xue, Dongling [1 ]
Lin, Xiangyuan [1 ]
Hui, Meng [1 ]
机构
[1] Changan Univ, Sch Elect & Control Engn, Xian 710064, Peoples R China
关键词
Marine vehicles; Feature extraction; Detectors; Synthetic aperture radar; Semantics; Scattering; Radar polarimetry; Attention mechanism; convolutional neural networks (CNNs); feature balance; keypoint; ship detection; synthetic aperture radar (SAR);
D O I
10.1109/LGRS.2023.3252590
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
Most synthetic aperture radar (SAR) ship detectors based on convolutional neural networks (CNNs) needed preset anchor boxes to object classification and bounding box coordinate regression. However, the sparsity and unbalanced distribution of ships in SAR images mean that most anchor boxes are redundant. Thus, the anchor settings directly affect the performance and generalization ability of the detector. In addition, a variety in ship scales and the substantial interference of inshore backgrounds bring significant challenges to the SAR ship detector's performance improvement. In this letter, a novel anchor-free-based detector, named feature balance and united attention (FBUA-Net), is proposed. We adopt a keypoint-based strategy to predict bounding boxes to eliminate the influence of anchors. Besides, we propose a global context-guided feature balanced pyramid (GC-FBP), which balances the semantic information at different levels of the feature pyramid by aggregation and averaging and uses the global context module (GCM) to learn global contextual information to construct long-range dependencies between ship targets and the background. Considering the interference of scattering noise to the detector, a united attention module (UAM) is designed to reduce the interference of surrounding noise by focusing on the spatial shape and scale size of ship targets in both the spatial and scale domains. Experimental results on the SAR ship detection dataset (SSDD) and high-resolution SAR images dataset (HRSID) datasets show that our detector achieves state-of-the-art (SOTA) performance.
引用
收藏
页数:5
相关论文
共 15 条
[1]   GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond [J].
Cao, Yue ;
Xu, Jiarui ;
Lin, Stephen ;
Wei, Fangyun ;
Hu, Han .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION WORKSHOPS (ICCVW), 2019, :1971-1980
[2]   Dynamic Head: Unifying Object Detection Heads with Attentions [J].
Dai, Xiyang ;
Chen, Yinpeng ;
Xiao, Bin ;
Chen, Dongdong ;
Liu, Mengchen ;
Yuan, Lu ;
Zhang, Lei .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :7369-7378
[3]   CenterNet: Keypoint Triplets for Object Detection [J].
Duan, Kaiwen ;
Bai, Song ;
Xie, Lingxi ;
Qi, Honggang ;
Huang, Qingming ;
Tian, Qi .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :6568-6577
[4]   An Anchor-Free Method Based on Feature Balancing and Refinement Network for Multiscale Ship Detection in SAR Images [J].
Fu, Jiamei ;
Sun, Xian ;
Wang, Zhirui ;
Fu, Kun .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (02) :1331-1344
[5]   Enriching SAR Ship Detection via Multistage Domain Alignment [J].
Jeong, Somi ;
Kim, Youngjung ;
Kim, Sungho ;
Sohn, Kwanghoon .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
[6]   FoveaBox: Beyound Anchor-Based Object Detection [J].
Kong, Tao ;
Sun, Fuchun ;
Liu, Huaping ;
Jiang, Yuning ;
Li, Lei ;
Shi, Jianbo .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2020, 29 :7389-7398
[7]   A Lightweight Faster R-CNN for Ship Detection in SAR Images [J].
Li, Yiding ;
Zhang, Shunsheng ;
Wang, Wen-Qin .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
[8]   Feature Pyramid Networks for Object Detection [J].
Lin, Tsung-Yi ;
Dollar, Piotr ;
Girshick, Ross ;
He, Kaiming ;
Hariharan, Bharath ;
Belongie, Serge .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :936-944
[9]  
Liu H., 2021, arXiv
[10]   Libra R-CNN: Towards Balanced Learning for Object Detection [J].
Pang, Jiangmiao ;
Chen, Kai ;
Shi, Jianping ;
Feng, Huajun ;
Ouyang, Wanli ;
Lin, Dahua .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :821-830