Electronics Optics & Control, Volume. 30, Issue 5, 99(2023)
Lightweight Ship Target Detection Algorithm for SAR Images
Accurate ship target detection technology can improve the omni-directional perception ability of weapons and equipment.Aiming at the serious problem of false alarm and missing alarm in SAR ship target detection in complex environment,a ship target detection algorithm 3S-YOLO based on YOLOv5 in lightweight SAR image is proposed.Firstly,3S-YOLO algorithm reconstructs the network structure,adjusts the relationship between receptive field and multi-scale fusion,and realizes the lightweight processing of feature extraction network and feature fusion network.Then,the network is pruned,and compressed by FPGM pruning algorithm to speed up the reasoning.Finally,the network is trained with varifocal loss to make IACS regression.The results show that the accuracy of the algorithm can be improved to 99.1% after optimization.After pruning,the volume of the model is greatly reduced,which can be compressed to 190 kiB,a decrease of 98.6%.The reasoning speed of the algorithm is increased by 4 times,and the reasoning time is reduced to less than 3 ms.Compared with the current mainstream algorithms,3S-YOLO has achieved good results in all aspects,which can meet the real-time ship target detection in SAR images.
Get Citation
Copy Citation Text
WANG Hengtao, ZHANG Shang. Lightweight Ship Target Detection Algorithm for SAR Images[J]. Electronics Optics & Control, 2023, 30(5): 99
Category:
Received: Apr. 18, 2021
Accepted: --
Published Online: Nov. 29, 2023
The Author Email: