A Lightweight Convolutional Neural Network for Ship Target Detection in SAR Images

被引:8
作者
Hao, Yisheng [1 ]
Zhang, Ying [2 ]
机构
[1] Shanghai Maritime Univ, Inst Logist Sci & Engn, Shanghai 201306, Peoples R China
[2] Shanghai Maritime Univ, Fac Coll Informat Engn, Shanghai 201306, Peoples R China
基金
中国国家自然科学基金;
关键词
Object detection; Radar polarimetry; Marine vehicles; Classification algorithms; Feature extraction; Convolutional neural networks; Clutter; Anchor free; lightweight convolutional neural network; synthetic aperture radar (SAR); target detection; CFAR DETECTION;
D O I
10.1109/TAES.2023.3344396
中图分类号
V [航空、航天];
学科分类号
08 ; 0825 ;
摘要
Synthetic aperture radar (SAR), due to its merits of all-day, all-weather, and high resolution, has become an important component of radar research. Target detection is a significant basis for radar image interpretation. Aiming at the problems, such as low accuracy and high complexity of traditional convolutional neural network in SAR ship target detection, which is not conducive to the deployment of terminal equipment, we proposed a lightweight network for SAR image ship target detection based on the latest anchor-free algorithm YOLOX. First, we redesign a novel lightweight block with stronger feature fusion ability, namely, MobileNetV3S. On the basis of MobileNetV3S, we combine Cross Stage Partial Network to construct a lightweight backbone. Second, in order to improve the ability of multiscale feature extraction, we propose a new module based on dilated convolution with different dilated rates and Efficient Spatial Pyramid Network. Furthermore, we adopt convolutional block attentional module to optimize traditional YOLOX's Feature Pyramid Network, and propose a lightweight enhanced feature extraction module, which can improve the focusing ability of important targets. For detection head, the depth-separable convolution is also applied to reduce the network's parameters. Finally, in terms of loss function, we abandon the traditional Intersection over Union and use absolute Intersection over Union with the better convergence effect. The experimental results on the SAR Ship Detection Dataset show that compared with the baseline YOLOX, although parameters of our method are decreased by 66.7%, its AP reaches 90.8%, which exceeds the baseline YOLOX by 0.5%, and its false detection rate is also obviously reduced, achieving state-of-the-art performance.
引用
收藏
页码:1882 / 1898
页数:17
相关论文
共 56 条
[1]  
Alkhaleefah Mohammad, 2021, 2021 IEEE International Geoscience and Remote Sensing Symposium IGARSS, P3030, DOI 10.1109/IGARSS47720.2021.9553632
[2]   Learning Slimming SAR Ship Object Detector Through Network Pruning and Knowledge Distillation [J].
Chen, Shiqi ;
Zhan, Ronghui ;
Wang, Wei ;
Zhang, Jun .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2021, 14 :1267-1282
[3]   Anchor-free Convolutional Network with Dense Attention Feature Aggregation for Ship Detection in SAR Images [J].
Gao, Fei ;
He, Yishan ;
Wang, Jun ;
Hussain, Amir ;
Zhou, Huiyu .
REMOTE SENSING, 2020, 12 (16)
[4]  
Ge Z, 2021, Arxiv, DOI [arXiv:2107.08430, 10.48550/arXiv.2107.08430, DOI 10.48550/ARXIV.2107.08430]
[5]  
Girshick R, 2015, Arxiv, DOI arXiv:1504.08083
[6]   Rich feature hierarchies for accurate object detection and semantic segmentation [J].
Girshick, Ross ;
Donahue, Jeff ;
Darrell, Trevor ;
Malik, Jitendra .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :580-587
[7]   A review of semantic segmentation using deep neural networks [J].
Guo, Yanming ;
Liu, Yu ;
Georgiou, Theodoros ;
Lew, Michael S. .
INTERNATIONAL JOURNAL OF MULTIMEDIA INFORMATION RETRIEVAL, 2018, 7 (02) :87-93
[8]  
Han K, 2020, Arxiv, DOI arXiv:1911.11907
[9]   Multilayer CFAR Detection of Ship Targets in Very High Resolution SAR Images [J].
Hou, Biao ;
Chen, Xingzhong ;
Jiao, Licheng .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2015, 12 (04) :811-815
[10]  
Howard A. G., 2017, ARXIV170404861