YOLOSAR-Lite: a lightweight framework for real-time ship detection in SAR imagery

被引:0
|
作者
Wang, Haochen [1 ]
Shi, Juan [1 ]
Karimian, Hamed [1 ]
Liu, Fucheng [1 ]
Wang, Fei [1 ]
机构
[1] Jiangsu Ocean Univ, Sch Marine Technol & Geomat, 311,59 Cangwu Rd, Lianyungang 222005, Haizhou, Peoples R China
基金
中国国家自然科学基金;
关键词
Synthetic aperture radar (SAR); YOLOv8; lightweight model; ship recognition; knowledge distillation; NEURAL-NETWORK; ALGORITHM; DATASET; MODEL;
D O I
10.1080/17538947.2024.2405525
中图分类号
P9 [自然地理学];
学科分类号
0705 ; 070501 ;
摘要
Real-time ship detection using SAR images is crucial for maritime operations but remains challenging due to issues such as sidelobes, complex background interferences, and target defocusing. Studies have shown that deep learning can address these problems, current deep learning algorithms often suffer from achieving the necessary accuracy due to complex noise and background characteristics. Additionally, advanced algorithms excelling in feature extraction usually require significant computational resources. This paper proposes a novel deep learning approach based on YOLOv8 to improve accuracy while reducing model complexity and computational burden. The approach innovatively integrates a knowledge distillation module into the YOLOv8. We further enhance the model by replacing the original YOLOv8 backbone, neck, and head with lightweight alternatives: HGNetv2, SlimNeck, and a newly designed decoupled head. The enhanced model, YOLOSAR-Lite, demonstrated exceptional performance on both accuracy and model complexity metrics. Rigorous evaluation on the Official-SSDD dataset and the SAR-Ship-Dataset revealed recognition accuracies of 95.32% and 96.06%, respectively. Furthermore, YOLOSAR-Lite achieved significant reduction in complexity compared to the YOLOv8: 4.48G FLOPs (45.30% reduction), 2.05M parameters (31.89% reduction), and 3.91MB model size (31.76% reduction). These improvements demonstrate YOLOSAR-Lite achieves higher accuracy with minimal complexity. Consequently, YOLOSAR-Lite is a promising solution for real-time ship detection.
引用
收藏
页数:23
相关论文
共 50 条
  • [31] Towards real-time detection of landmines in FLIR imagery
    Ito, MR
    Duong, S
    McFee, JE
    Russell, KL
    2001 IEEE WORKSHOP ON STATISTICAL SIGNAL PROCESSING PROCEEDINGS, 2001, : 154 - 157
  • [32] Real-time Vehicle Detection from UAV Imagery
    Xie, Xuemei
    Yang, Wenzhe
    Cao, Guimei
    Yang, Jianxiu
    Zhao, Zhifu
    Chen, Shu
    Liao, Quan
    Shi, Guangming
    2018 IEEE FOURTH INTERNATIONAL CONFERENCE ON MULTIMEDIA BIG DATA (BIGMM), 2018,
  • [33] A waterborne salient ship detection method on SAR imagery
    MA Long
    CHEN Liang
    ZHANG XueJing
    CHEN He
    Nouman Qadeer SOOMRO
    Science China(Information Sciences), 2015, 58 (08) : 193 - 196
  • [34] A waterborne salient ship detection method on SAR imagery
    Ma Long
    Chen Liang
    Zhang XueJing
    Chen He
    Soomro, Nouman Qadeer
    SCIENCE CHINA-INFORMATION SCIENCES, 2015, 58 (08) : 1 - 3
  • [35] A waterborne salient ship detection method on SAR imagery
    Long Ma
    Liang Chen
    XueJing Zhang
    He Chen
    Nouman Qadeer Soomro
    Science China Information Sciences, 2015, 58 : 1 - 3
  • [36] An Adaptive Ship Detection Scheme for Spaceborne SAR Imagery
    Leng, Xiangguang
    Ji, Kefeng
    Zhou, Shilin
    Xing, Xiangwei
    Zou, Huanxin
    SENSORS, 2016, 16 (09):
  • [37] Comparison of ship detection algorithms in spaceborne SAR imagery
    Chen, P
    Huang, WG
    Yang, JS
    Fu, B
    Lou, XL
    Shi, AQ
    IGARSS 2005: IEEE International Geoscience and Remote Sensing Symposium, Vols 1-8, Proceedings, 2005, : 1750 - 1752
  • [38] An improved CFAR model for ship detection in SAR imagery
    Huang, WG
    Chen, P
    Yang, JS
    Fu, B
    Xiao, QM
    Yao, L
    Zhou, CB
    IGARSS 2004: IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM PROCEEDINGS, VOLS 1-7: SCIENCE FOR SOCIETY: EXPLORING AND MANAGING A CHANGING PLANET, 2004, : 4719 - 4722
  • [39] Ship Detection in SAR Imagery Based on Density and Clustering
    Hao, Mengxi
    Luo, Yang
    Zhai, Wenjing
    Jin, Songzhi
    2020 CHINESE AUTOMATION CONGRESS (CAC 2020), 2020, : 6974 - 6977
  • [40] Significance based Ship Detection from SAR Imagery
    Arivazhagan, S.
    Jebarani, W. Sylvia Lilly
    Shebiah, R. Newlin
    Ligi, S. Vineth
    Kumar, P. V. Hareesh
    Anilkumar, K.
    PROCEEDINGS OF 2019 1ST INTERNATIONAL CONFERENCE ON INNOVATIONS IN INFORMATION AND COMMUNICATION TECHNOLOGY (ICIICT 2019), 2019,