• Electronics Optics & Control
  • Vol. 30, Issue 12, 66 (2023)
ZHANG Shang1、2、3, CHEN Yifang1、2、3, WANG Shentao4, WANG Hengtao3、5, and RAN Xiukang1、2、3
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • 3[in Chinese]
  • 4[in Chinese]
  • 5[in Chinese]
  • show less
    DOI: 10.3969/j.issn.1671-637x.2023.12.011 Cite this Article
    ZHANG Shang, CHEN Yifang, WANG Shentao, WANG Hengtao, RAN Xiukang. An Improved Ship Target Detection Algorithm Based on YOLOv5[J]. Electronics Optics & Control, 2023, 30(12): 66 Copy Citation Text show less

    Abstract

    The ship has become an important monitoring target in maritime military field.Ship target detection in SAR images suffers from poor detection effects, large computation amount and weak generalization capability.To solve the problems, a lightweight ship target detection algorithm based on YOLOv5 and Mobilenetv3 is proposed.Firstly, Mobilenetv3 backbone network is introduced to reduce the computation amount and volume of the model and realize lightweight processing of the model.Then, the EIoU loss function is introduced to improve the regression accuracy and convergence speed of the prediction box.Finally, CBAM is introduced into the neck network, and attention adjustment is conducted at the stage of feature fusion to improve the detection accuracy and detection effects of the model.The experimental results on SSDD dataset show that the volume of the improved algorithm model is reduced to 18.32% of that of the original YOLOv5 model, the training time is shortened by 35.22%, the parameter quantity is reduced to 15.94% of that of the original model, the computation amount is reduced to 10.76% of that of the original model, and mAP is improved to 98.3%.The experimental results show that the improved algorithm greatly reduces parameter quantity, computation amount, model volume and training time while maintaining high-precision detection effects, which can realize real-time detection of ship targets in SAR images.
    ZHANG Shang, CHEN Yifang, WANG Shentao, WANG Hengtao, RAN Xiukang. An Improved Ship Target Detection Algorithm Based on YOLOv5[J]. Electronics Optics & Control, 2023, 30(12): 66
    Download Citation