Chinese Journal of Ship Research, Volume. 19, Issue 5, 200(2024)

Lightweight remote sensing ship detection algorithm based on YOLOv5s

Haochen WANG... Yuelan XIN, Jiang GUO and Qingqing WANG |Show fewer author(s)
Author Affiliations
  • College of Physics and Electronic Information Engineering, Qinghai Normal University, Xining 810001, China
  • show less

    Objective

    This paper proposes a lightweight remote sensing ship target detection algorithm LR-YOLO based on improved YOLOv5s to meet the lightweight and fast inference requirements of ship target detection tasks involving remote sensing images.

    Methods

    First, the backbone network adopts the ShuffleNet v2 block stacking method, effectively reducing the number of network model parameters and improving the computational speed; second, a region selection module filter is designed to select regions of interest and extract effective features more fully; finally, a circular smooth label is introduced to calculate angle loss and perform rotation detection on remote sensing ship targets, while deformable convolution is used to adapt to geometric deformation and improve detection performance.

    Results

    The experimental results on the HRSC2016 ship dataset show that the detection accuracy of the algorithm reaches 92.90%, an improvement of 1.3%, with the number of network model parameters only 39.33% that of the baseline model.

    Conclusion

    The proposed algorithm achieves a balance between lightweight and detection accuracy, providing references for remote sensing ship target detection.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Haochen WANG, Yuelan XIN, Jiang GUO, Qingqing WANG. Lightweight remote sensing ship detection algorithm based on YOLOv5s[J]. Chinese Journal of Ship Research, 2024, 19(5): 200

    Download Citation

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

    Category: Weapon, Electronic and Information System

    Received: Jul. 10, 2023

    Accepted: --

    Published Online: Mar. 14, 2025

    The Author Email:

    DOI:10.19693/j.issn.1673-3185.03454

    Topics