Ship Detection in Synthetic Aperture Radar Images Based on BiLevel Spatial Attention and Deep Poly Kernel Network

被引:2
作者
Tian, Siyuan [1 ]
Jin, Guodong [1 ]
Gao, Jing [1 ]
Tan, Lining [1 ]
Xue, Yuanliang [1 ]
Li, Yang [1 ]
Liu, Yantong [2 ]
机构
[1] Xian Res Inst Hitech, Xian 710025, Peoples R China
[2] Kunsan Natl Univ, Dept Comp & Informat Engn, Gunsan 54150, South Korea
基金
中国国家自然科学基金;
关键词
SAR; ship detection; multiscale; deep learning; YOLO;
D O I
10.3390/jmse12081379
中图分类号
U6 [水路运输]; P75 [海洋工程];
学科分类号
0814 ; 081505 ; 0824 ; 082401 ;
摘要
Synthetic aperture radar (SAR) is a technique widely used in the field of ship detection. However, due to the high ship density, fore-ground-background imbalance, and varying target sizes, achieving lightweight and high-precision multiscale ship object detection remains a significant challenge. In response to these challenges, this research presents YOLO-MSD, a multiscale SAR ship detection method. Firstly, we propose a Deep Poly Kernel Backbone Network (DPK-Net) that utilizes the Optimized Convolution (OC) Module to reduce data redundancy and the Poly Kernel (PK) Module to improve the feature extraction capability and scale adaptability. Secondly, we design a BiLevel Spatial Attention Module (BSAM), which consists of the BiLevel Routing Attention (BRA) and the Spatial Attention Module. The BRA is first utilized to capture global information. Then, the Spatial Attention Module is used to improve the network's ability to localize the target and capture high-quality detailed information. Finally, we adopt a Powerful-IoU (P-IoU) loss function, which can adjust to the ship size adaptively, effectively guiding the anchor box to achieve faster and more accurate detection. Using HRSID and SSDD as experimental datasets, mAP of 90.2% and 98.8% are achieved, respectively, outperforming the baseline by 5.9% and 6.2% with a model size of 12.3 M. Furthermore, the network exhibits excellent performance across various ship scales.
引用
收藏
页数:22
相关论文
共 50 条
[1]   Feature Enhancement Pyramid and Shallow Feature Reconstruction Network for SAR Ship Detection [J].
Bai, Lin ;
Yao, Cheng ;
Ye, Zhen ;
Xue, Dongling ;
Lin, Xiangyuan ;
Hui, Meng .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2023, 16 :1042-1056
[2]  
Bochkovskiy A, 2020, Arxiv, DOI arXiv:2004.10934
[3]  
Cai XH, 2024, Arxiv, DOI [arXiv:2403.06258, 10.48550/ARXIV.2403.06258, 10.48550/arXiv.2403.06258 2403.06258]
[4]   Cascade R-CNN: Delving into High Quality Object Detection [J].
Cai, Zhaowei ;
Vasconcelos, Nuno .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :6154-6162
[5]   MLSDNet: Multiclass Lightweight SAR Detection Network Based on Adaptive Scale Distribution Attention [J].
Chang, Hao ;
Fu, Xiongjun ;
Dong, Jian ;
Liu, Jiaang ;
Zhou, Zixiang .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2023, 20
[6]   Run, Don't Walk: Chasing Higher FLOPS for Faster Neural Networks [J].
Chen, Jierun ;
Kao, Shiu-Hong ;
He, Hao ;
Zhuo, Weipeng ;
Wen, Song ;
Lee, Chul-Ho ;
Chan, S. -H. Gary .
2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, :12021-12031
[7]   Regional attention-based single shot detector for SAR ship detection [J].
Chen Shiqi ;
Zhan Ronghui ;
Zhang Jun .
JOURNAL OF ENGINEERING-JOE, 2019, 2019 (21) :7381-7384
[8]   Multi-Scale Ship Detection Algorithm Based on YOLOv7 for Complex Scene SAR Images [J].
Chen, Zhuo ;
Liu, Chang ;
Filaretov, V. F. ;
Yukhimets, D. A. .
REMOTE SENSING, 2023, 15 (08)
[9]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[10]  
Cordonnier JB, 2021, Arxiv, DOI [arXiv:2006.16362, DOI 10.48550/ARXIV.2006.16362]