LH-YOLO: A Lightweight and High-Precision SAR Ship Detection Model Based on the Improved YOLOv8n

被引:1
作者
Cao, Qi [1 ]
Chen, Hang [1 ]
Wang, Shang [1 ]
Wang, Yongqiang [1 ]
Fu, Haisheng [1 ]
Chen, Zhenjiao [2 ]
Liang, Feng [1 ]
机构
[1] Xi An Jiao Tong Univ, Sch Microelect, Xian 710049, Peoples R China
[2] Northwestern Polytech Univ, Sch Elect & Informat, Xian 710072, Peoples R China
基金
中国国家自然科学基金;
关键词
lightweight network; synthetic aperture radar (SAR); feature fusion; shared detection head; YOLOv8n; weight sharing mechanism; OS-CFAR;
D O I
10.3390/rs16224340
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
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.
引用
收藏
页数:21
相关论文
共 55 条
  • [1] OS-CFAR and CMLD threshold optimization in distributed systems using evolutionary strategies
    Abdou, Latifa
    Soltani, Faouzi
    [J]. SIGNAL IMAGE AND VIDEO PROCESSING, 2008, 2 (02) : 155 - 167
  • [2] Adarsh P, 2020, INT CONF ADVAN COMPU, P687, DOI [10.1109/ICACCS48705.2020.9074315, 10.1109/icaccs48705.2020.9074315]
  • [3] Agrawal S., 2019, The International Archives of the Photogrammetry, VXLII-5-W3, P1, DOI DOI 10.5194/ISPRS-ARCHIVES-XLII-5-W3-1-2019
  • [4] Ship Detection With SAR C-Band Satellite Images: A Systematic Review
    Alexandre, Cyprien
    Devillers, Rodolphe
    Mouillot, David
    Seguin, Raphael
    Catry, Thibault
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2024, 17 : 14353 - 14367
  • [5] Mixture-Based Superpixel Segmentation and Classification of SAR Images
    Arisoy, Sertac
    Kayabol, Koray
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2016, 13 (11) : 1721 - 1725
  • [7] Cascade R-CNN: Delving into High Quality Object Detection
    Cai, Zhaowei
    Vasconcelos, Nuno
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 6154 - 6162
  • [8] Chen SZ, 2014, 2014 INTERNATIONAL CONFERENCE ON DATA SCIENCE AND ADVANCED ANALYTICS (DSAA), P541, DOI 10.1109/DSAA.2014.7058124
  • [9] Convolutional Neural Network With Data Augmentation for SAR Target Recognition
    Ding, Jun
    Chen, Bo
    Liu, Hongwei
    Huang, Mengyuan
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2016, 13 (03) : 364 - 368
  • [10] Ship detection with deep learning: a survey
    Er, Meng Joo
    Zhang, Yani
    Chen, Jie
    Gao, Wenxiao
    [J]. ARTIFICIAL INTELLIGENCE REVIEW, 2023, 56 (10) : 11825 - 11865