Electronics Optics & Control, Volume. 31, Issue 8, 32(2024)
An Improved Algorithm for Detecting Ship Target in SAR Images Based on YOLOv5 Model
Aiming at the problems of poor detection accuracy and large amount of computation in the existing SAR ship target detection methods,a lightweight ship target detection method based on YOLOv5 and GhostNet is proposed.The GhostConv and GhostC3 modules of the lightweight network GhostNet are introduced to improve the backbone network of YOLOv5,achieving a significant reduction in model computation.The CBAMC3 module is introduced in the neck network to adjust attention during the feature fusion stage and achieve accurate target detection.In addition,the EIoU loss function is introduced to improve the regression accuracy and rate of convergence of the prediction box.The test results on the public dataset indicate that the improved algorithm significantly reduces the number of parameters and model volume while maintaining high accuracy,making it an ideal lightweight ship detection model for SAR images.
Get Citation
Copy Citation Text
SUN Peishuang, WEN Xianbin. An Improved Algorithm for Detecting Ship Target in SAR Images Based on YOLOv5 Model[J]. Electronics Optics & Control, 2024, 31(8): 32
Received: Jul. 5, 2023
Accepted: --
Published Online: Oct. 22, 2024
The Author Email: