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 条
  • [41] 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
  • [42] 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,
  • [43] A real-time framework for eye detection and tracking
    Hussein O. Hamshari
    Steven S. Beauchemin
    Journal of Real-Time Image Processing, 2011, 6 : 235 - 245
  • [44] Real-time IRC Threat Detection Framework
    Shao, Sicong
    Tunc, Cihan
    Satam, Pratik
    Hariri, Salim
    2017 IEEE 2ND INTERNATIONAL WORKSHOPS ON FOUNDATIONS AND APPLICATIONS OF SELF* SYSTEMS (FAS*W), 2017, : 318 - 323
  • [45] A real-time framework for eye detection and tracking
    Hamshari, Hussein O.
    Beauchemin, Steven S.
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2011, 6 (04) : 235 - 245
  • [46] Spade: A Real-Time Fraud Detection Framework
    Jiang, Jiaxin
    Zhang, Zhen
    Luo, Bingqiao
    He, Bingsheng
    Chen, Min
    Wang, Weiyang
    Chen, Jia
    PROCEEDINGS OF THE VLDB ENDOWMENT, 2024, 17 (12): : 4253 - 4256
  • [47] A Framework for Real-Time Spam Detection in Twitter
    Gupta, Himank
    Jamal, Mohd. Saalim
    Madisetty, Sreekanth
    Desarkar, Maunendra Sankar
    2018 10TH INTERNATIONAL CONFERENCE ON COMMUNICATION SYSTEMS & NETWORKS (COMSNETS), 2018, : 380 - 387
  • [48] A fast target detection framework for SAR imagery
    He, Zhigiang
    Pei, Jifang
    Yang, Haiguang
    Huang, Yulin
    Yang, Jianyu
    2016 CIE INTERNATIONAL CONFERENCE ON RADAR (RADAR), 2016,
  • [49] A Lightweight Model for Real-Time Detection of Vehicle Black Smoke
    Chen, Ke
    Wang, Han
    Zhai, Yingchao
    SENSORS, 2023, 23 (23)
  • [50] Accurate and Lightweight RailNet for Real-Time Rail Line Detection
    Tao, Zhen
    Ren, Shiwei
    Shi, Yueting
    Wang, Xiaohua
    Wang, Weijiang
    ELECTRONICS, 2021, 10 (16)