Electronics Optics & Control, Volume. 31, Issue 8, 32(2024)

An Improved Algorithm for Detecting Ship Target in SAR Images Based on YOLOv5 Model

SUN Peishuang and WEN Xianbin
Author Affiliations
  • [in Chinese]
  • show less

    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.

    Tools

    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

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Received: Jul. 5, 2023

    Accepted: --

    Published Online: Oct. 22, 2024

    The Author Email:

    DOI:10.3969/j.issn.1671-637x.2024.08.005

    Topics