Electronics Optics & Control, Volume. 31, Issue 5, 46(2024)

An Improved YOLOv7 Based Algorithm for Ship Target Detection in SAR Images

ZHANG Shang1...2,3, LI Mengsi1,2,3, CHEN Yonglin1,2,3, and ZHANG Zhuo1,23 |Show fewer author(s)
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • 3[in Chinese]
  • show less

    In SAR ship dataset,small objects only account for a small proportion of the images pixels,the objects cannot be clearly recognized,and the detection efficiency is low.To solve the problems,an improved YOLOv7 based SAR ship target detection algorithm STSD-YOLO is proposed.Firstly,according to the characteristics of SAR images,the network structure is redesigned,and the relationship between multi-scale feature fusion and feature extraction is changed to solve the problem of losing detailed features due to excessive down-sampling times.Secondly,a lightweight attention mechanism named Shuffle Attention is used to fuse feature grouping with channel replacement based on the spatial domain and channel domain attention mechanism to improve the feature extraction ability of the network and reduce computational complexity.Then,the convolution variant DSConv is introduced to reduce computational intensity by storing only integers in the variable quantization kernel.Finally,the NWD metric is added to improve the performance of small object detection by modeling the bounding box as a 2D Gaussian distribution to measure the similarity between the bounding boxes of small objects.The HRSID ship dataset is adopted for experimental verification.In comparison with the baseline algorithm,the STSD-YOLO algorithm has its mAP increased by 9.9%in the ship detection task,and model volume reduced by 62.55%.Through comparative experiments,it is shown that the improved algorithm has better detection effects than other mainstream algorithms.It can effectively address the difficulties of SAR image detection,which is competent to carry out the ship detection task in SAR images.

    Tools

    Get Citation

    Copy Citation Text

    ZHANG Shang, LI Mengsi, CHEN Yonglin, ZHANG Zhuo. An Improved YOLOv7 Based Algorithm for Ship Target Detection in SAR Images[J]. Electronics Optics & Control, 2024, 31(5): 46

    Download Citation

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

    Category:

    Received: Jun. 29, 2023

    Accepted: --

    Published Online: Aug. 23, 2024

    The Author Email:

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

    Topics