Electronics Optics & Control, Volume. 31, Issue 12, 19(2024)

A Ship Target Detection Method for SAR images Based on Improved YOLOv7 Algorithm

CHEN Wenhan1... ZHU Zhengwei1 and SONG Changlong2 |Show fewer author(s)
Author Affiliations
  • 1School of Information Engineering, Southwest University of Science and Technology, Mianyang 621000, China
  • 2China Ordnance Equipment Group Automation Research Institute Co. Ltd., Mianyang 621000, China
  • show less

    In order to solve the problems of low target detection accuracy and difficulty in small target detection caused by blurred images and lack of texture features in the existing mainstream algorithms for ship target detection in SAR images, and considering that the real-time performance of the network will be affected by introducing too many parameters, an improved YOLOv7 ship target detection method based on coordinate attention mechanism and Normalized Wasserstein Distance (NWD) metric is proposed. Firstly, the maximum pooling and residual structure are introduced into the coordinate attention mechanism to improve the model feature extraction ability. Secondly, combining dense connection with lightweight convolution, SPPCSPC-P is designed to enhance the fusion between features. In addition, a small target detection layer is added to the backbone network to improve the low detection accuracy of the model for small targets. Finally, the weighted positioning loss function is designed by using NWD metric and CIoU loss, which further improves the model detection accuracy. Experiments are carried out on SSDD dataset, and the experimental results show that the average accuracy of this method reaches 98.38%, which is 2.09 percentage points higher than that of YOLOv7 network.

    Tools

    Get Citation

    Copy Citation Text

    CHEN Wenhan, ZHU Zhengwei, SONG Changlong. A Ship Target Detection Method for SAR images Based on Improved YOLOv7 Algorithm[J]. Electronics Optics & Control, 2024, 31(12): 19

    Download Citation

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

    Category:

    Received: Dec. 5, 2023

    Accepted: Dec. 25, 2024

    Published Online: Dec. 25, 2024

    The Author Email:

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

    Topics