LH-YOLO: A Lightweight and High-Precision SAR Ship Detection Model Based on the Improved YOLOv8n
Synthetic aperture radar is widely applied to ship detection due to generating high-resolution images under diverse weather conditions and its penetration capabilities, making SAR images a valuable data source. However, detecting multi-scale ship targets in complex backgrounds leads to issues of false positives and missed detections, posing challenges for lightweight and high-precision algorithms. There is an urgent need to improve accuracy of algorithms and their deployability. This paper introduces LH-YOLO, a YOLOv8n-based, lightweight, and high-precision SAR ship detection model. We propose a lightweight backbone network, StarNet-nano, and employ element-wise multiplication to construct a lightweight feature extraction module, LFE-C2f, for the neck of LH-YOLO. Additionally, a reused and shared convolutional detection (RSCD) head is designed using a weight sharing mechanism. These enhancements significantly reduce model size and computational demands while maintaining high precision. LH-YOLO features only 1.862 M parameters, representing a 38.1% reduction compared to YOLOv8n. It exhibits a 23.8% reduction in computational load while achieving a mAP50 of 96.6% on the HRSID dataset, which is 1.4% higher than YOLOv8n. Furthermore, it demonstrates strong generalization on the SAR-Ship-Dataset with a mAP50 of 93.8%, surpassing YOLOv8n by 0.7%. LH-YOLO is well-suited for environments with limited resources, such as embedded systems and edge computing platforms.
Authors
- Haisheng Fu (ORCID: https://orcid.org/0000-0002-0113-5500)
- Zhenjiao Chen
- Shang Wang
- Feng Liang (ORCID: https://orcid.org/0000-0002-9393-6224)
- Hang Chen (ORCID: https://orcid.org/0000-0002-9011-4110)
- Qi Cao (ORCID: https://orcid.org/0009-0002-1244-4568)
- Yongqiang Wang (ORCID: https://orcid.org/0000-0002-7276-4275)
Institutions
- Northwestern Polytechnical University (CN)
- Xi'an Jiaotong University (CN)
Publication Details
- Journal
- Remote Sensing
- Published
- 2024-11-20
- DOI
- https://doi.org/10.3390/rs16224340
- Citations
- 18
- Primary Topic
- Synthetic Aperture Radar (SAR) Applications and Techniques
- Type
- article
- Field-Weighted Citation Impact
- 23.38
Funders
- National Natural Science Foundation of China
- Chinese Aeronautical Establishment