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] 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
  • [42] 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
  • [43] 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
  • [44] 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
  • [45] 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,
  • [46] A Lightweight Model for Real-Time Detection of Vehicle Black Smoke
    Chen, Ke
    Wang, Han
    Zhai, Yingchao
    SENSORS, 2023, 23 (23)
  • [47] Accurate and Lightweight RailNet for Real-Time Rail Line Detection
    Tao, Zhen
    Ren, Shiwei
    Shi, Yueting
    Wang, Xiaohua
    Wang, Weijiang
    ELECTRONICS, 2021, 10 (16)
  • [48] Real-Time Hand Detection Method Based on Lightweight Network
    Jin, Fangrui
    Wang, Yangping
    Yong, Jiu
    Computer Engineering and Applications, 2023, 59 (14) : 192 - 200
  • [49] A lightweight multi-target real-time detection model
    Qiu B.
    Liu X.
    Shi Y.
    Shang Y.
    Beijing Hangkong Hangtian Daxue Xuebao/Journal of Beijing University of Aeronautics and Astronautics, 2020, 46 (09): : 1778 - 1785
  • [50] WiHumo: a real-time lightweight indoor human motion detection
    Yang, Hao
    Xu, Hua
    Tang, Keming
    INTERNATIONAL JOURNAL OF SENSOR NETWORKS, 2017, 24 (02) : 110 - 117