Research on lightweight and feature enhancement of SAR image ship targets detection

被引:0
作者
Gong J. [1 ]
Fu W. [1 ]
Fang H. [2 ]
机构
[1] School of Communication Engineering, Xidian University, Xi'an
[2] School of Computer Science and Technology, Xidian University, Xi'an
来源
Xi'an Dianzi Keji Daxue Xuebao/Journal of Xidian University | 2024年 / 51卷 / 02期
关键词
convolutional neural networks; feature extraction; object detection; synthetic aperture radar;
D O I
10.19665/j.issn1001-2400.20230407
中图分类号
学科分类号
摘要
The accuracy of ship targets detection in sythetic aperture radar images is susceptible to the nearshore clutter. The existing detection algorithms are highly complex and difficult to deploy on embedded devices. Due to these problems a lightweight and high-precision SAR image ship target detection algorithm CA-Shuffle- YOLO (Coordinate Shuffle You Only Look Once) is proposed in this article. Based on the YOLO v5 target detection algorithm, the backbone network is improved in two aspects: lightweight and feature refinement. The lightweight module is introduced to reduce the computational complexity of the network and improve the reasoning speed, and a collaborative attention mechanism module is introduced to enhance the algorithm's ability to extract the detailed information on near-shore ship targets. In the feature fusion network, weighted feature fusion and cross-module fusion are used to enhance the ability of the model to fuse the detailed information on SAR ship targets. At the same time, the depth separable convolution is used to reduce the computational complexity and improve the real-time performance. Through the test and comparison experiments on the SSDD ship target detection dataset, the results show that the detection accuracy of CA-Shuffle-YOLO is 97. 4%, the detection frame rate is 206FPS, and the required computational complexity is 6. lGFlops. Compare to the original YOLO v5, the FPS of our algorithm is 60FPS higher with the required computational complexity of our algorithm being only the 12% that of the ordinary YOLOvS. © 2024 Science Press. All rights reserved.
引用
收藏
页码:96 / 106
页数:10
相关论文
共 18 条
[1]  
SU Juan, YANG Long, HUANG Hua, Et al., Improved SSD Algorithm for SAR Image Small Target Ship Detection [J], Journal of Systems Engineering and Electronics, 42, 5, pp. 1026-1034, (2020)
[2]  
HE Chu, ZHANG Yu, LIAO Zixian, Et al., SAR Image CFAR Target Detection Algorithm Based on Compressed Sensing, Geomatics and Information Science of Wuhan University, 39, 7, pp. 878-882, (2014)
[3]  
CUI Z, LI Q, CAO Z, Et al., Dense Attention Pyramid Networks for Multi-Scale Ship Detection in SAR Images [J], IEEE Transactions on Geoscience and Remote Sensing, 57, 11, pp. 8983-8997, (2019)
[4]  
CHEN S Q, ZHAN R H, ZHANG J., Robust Single Stage Detector Based on Two-Stage Regression for SAR Ship Detection [C], Proceedings of the 2nd International Conference on Innovation in Artificial Intelligence, pp. 169-174, (2018)
[5]  
WANG J, LIN Y, GUO J, Et al., SSS-YOLO: Towards More Accurate Detection for Small Ships in Sar Image, Remote Sensing Letters, 12, 2, pp. 93-102, (2021)
[6]  
REDMON J, FARHADI A., Yolov.3: An Incremental Improvement, (2018)
[7]  
JIANG J, FU X, QIN R, Et al., High-Speed Lightweight Ship Detection Algorithm Based on YOLO-V4 for Three-Channels RGB SAR Imaged, Remote Sensing, 13, 10, (2021)
[8]  
LI Yonggang, ZHU Weigang, HUANG Qiongnan, Et al., SAR Image Near-Shore Ship Target Detection Under Complex BackgroundD], Journal of Systems Engineering and Electronics, 44, 10, pp. 3096-3103, (2022)
[9]  
LI J, QU C, SHAO J., Ship Detection in SAR Images Based on an Improved Faster R-CNN [C], 2017 SAR in Big Data Era:Models, Methods and Applications, pp. 1-6, (2017)
[10]  
GUI Y, LI X, XUE L., A Multilayer Fusion Light-Head Detector for SAR Ship Detection, Sensors, 19, 5, (2019)