LssDet: A Lightweight Deep Learning Detector for SAR Ship Detection in High-Resolution SAR Images

被引:12
作者
Yan, Guoxu [1 ]
Chen, Zhihua [1 ]
Wang, Yi [1 ]
Cai, Yangwei [1 ]
Shuai, Shikang [1 ]
机构
[1] Nanjing Univ Sci & Technol, Natl Key Lab Transient Phys, Nanjing 210094, Peoples R China
关键词
synthetic aperture radar (SAR); ship detection; lightweight detector;
D O I
10.3390/rs14205148
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
Synthetic aperture radar (SAR) ship detection has been the focus of many previous studies. Traditional SAR ship detectors face challenges in complex environments due to the limitations of manual feature extraction. With the rise of deep learning (DL) techniques, SAR ship detection based on convolutional neural networks (CNNs) has achieved significant achievements. However, research on CNN-based SAR ship detection has mainly focused on improving detection accuracy, and relatively little research has been conducted on reducing computational complexity. Therefore, this paper proposes a lightweight detector, LssDet, for SAR ship detection. LssDet uses Shufflenet v2, YOLOX PAFPN and YOLOX Decopuled Head as the baseline networks, improving based on the cross sidelobe attention (CSAT) module, the lightweight path aggregation feature pyramid network (L-PAFPN) module and the Focus module. Specifically, the CSAT module is an attention mechanism that enhances the model's attention to the cross sidelobe region and models the long-range dependence between the channel and spatial information. The L-PAFPN module is a lightweight feature fusion network that achieves excellent performance with little computational effort and a low parametric count. The Focus module is a low-loss feature extraction structure. Experiments showed that on the Sar ship detection dataset(SSDD), LssDet's computational cost was 2.60 GFlops, the model's volume was 2.25 M and AP@[0.5:0.95] was 68.1%. On the Large-scale SAR ship detection dataset-v1.0 (LS-SSDD-v1.0), LssDet's computational cost was 4.49 GFlops, the model's volume was 2.25 M and AP@[0.5:0.95] was 27.8%. Compared to the baseline network, LssDet had a 3.6% improvement in AP@[0.5:0.95] on the SSDD, and LssDet had a 1.5% improvement in AP@[0.5:0.95] on the LS-SSDD-v1.0. At the same time, LssDet reduced Floating-point operations per second (Flops) by 7.1% and Paraments (Params) by 23.2%. Extensive experiments showed that LssDet achieves excellent detection results with minimal computational complexity. Furthermore, we investigated the effectiveness of the proposed module through ablation experiments.
引用
收藏
页数:19
相关论文
共 49 条
[1]  
[Anonymous], 2021, Ultralytics: YOLOv5
[2]  
[Anonymous], 2017, P IEEE C COMP VIS PA
[3]  
Bochkovskiy A., 2020, PREPRINT
[4]   Wide-Area Traffic Monitoring With the SAR/GMTI System PAMIR [J].
Cerutti-Maori, Delphine ;
Klare, Jens ;
Brenner, Andreas R. ;
Ender, Joachim H. G. .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2008, 46 (10) :3019-3030
[5]   Ship Detection Based on YOLOv2 for SAR Imagery [J].
Chang, Yang-Lang ;
Anagaw, Amare ;
Chang, Lena ;
Wang, Yi Chun ;
Hsiao, Chih-Yu ;
Lee, Wei-Hong .
REMOTE SENSING, 2019, 11 (07)
[6]   Hybrid Task Cascade for Instance Segmentation [J].
Chen, Kai ;
Pang, Jiangmiao ;
Wang, Jiaqi ;
Xiong, Yu ;
Li, Xiaoxiao ;
Sun, Shuyang ;
Feng, Wansen ;
Liu, Ziwei ;
Shi, Jianping ;
Ouyang, Wanli ;
Loy, Chen Change ;
Lin, Dahua .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :4969-4978
[7]   Ship Detection in Large-Scale SAR Images Via Spatial Shuffle-Group Enhance Attention [J].
Cui, Zongyong ;
Wang, Xiaoya ;
Liu, Nengyuan ;
Cao, Zongjie ;
Yang, Jianyu .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (01) :379-391
[8]   A Modified CFAR Algorithm Based on Object Proposals for Ship Target Detection in SAR Images [J].
Dai, Hui ;
Du, Lan ;
Wang, Yan ;
Wang, Zhaocheng .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2016, 13 (12) :1925-1929
[9]   A Parzen-Window-Kernel-Based CFAR Algorithm for Ship Detection in SAR Images [J].
Gao, Gui .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2011, 8 (03) :557-561
[10]   A High-Effective Implementation of Ship Detector for SAR Images [J].
Gao, S. ;
Liu, J. M. ;
Miao, Y. H. ;
He, Z. J. .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19