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 条
  • [1] YOLOSeaShip: a lightweight model for real-time ship detection
    Jiang, Xiaoliang
    Cai, Jianchen
    Wang, Ban
    EUROPEAN JOURNAL OF REMOTE SENSING, 2024, 57 (01)
  • [2] YOLO-Lite: An Efficient Lightweight Network for SAR Ship Detection
    Ren, Xiaozhen
    Bai, Yanwen
    Liu, Gang
    Zhang, Ping
    REMOTE SENSING, 2023, 15 (15)
  • [3] Algorithm/Hardware Codesign for Real-Time On-Satellite CNN-Based Ship Detection in SAR Imagery
    Yang, Geng
    Lei, Jie
    Xie, Weiying
    Fang, Zhenman
    Li, Yunsong
    Wang, Jiaxuan
    Zhang, Xin
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [4] Lightweight Deep Neural Networks for Ship Target Detection in SAR Imagery
    Wang, Jielei
    Cui, Zongyong
    Jiang, Ting
    Cao, Changjie
    Cao, Zongjie
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2023, 32 : 565 - 579
  • [5] Mixed YOLOv3-LITE: A Lightweight Real-Time Object Detection Method
    Zhao, Haipeng
    Zhou, Yang
    Zhang, Long
    Peng, Yangzhao
    Hu, Xiaofei
    Peng, Haojie
    Cai, Xinyue
    SENSORS, 2020, 20 (07)
  • [6] A real-time deep learning network for ship detection in SAR images
    Zhou, Wenxue
    Zhang, Huachun
    SIGNAL IMAGE AND VIDEO PROCESSING, 2024, 18 (02) : 1893 - 1899
  • [7] A real-time deep learning network for ship detection in SAR images
    Wenxue Zhou
    Huachun Zhang
    Signal, Image and Video Processing, 2024, 18 : 1893 - 1899
  • [8] LIGHTWEIGHT SAR SHIP DETECTION
    Sorensen, K. Aa
    Heiselberg, P.
    Heiselberg, H.
    IGARSS 2023 - 2023 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM, 2023, : 6430 - 6433
  • [9] LRTransDet: A Real-Time SAR Ship-Detection Network with Lightweight ViT and Multi-Scale Feature Fusion
    Feng, Kunyu
    Lun, Li
    Wang, Xiaofeng
    Cui, Xiaoxin
    REMOTE SENSING, 2023, 15 (22)
  • [10] A LIGHTWEIGHT NETWORK FOR MULTISCALE SAR SHIP DETECTION UNDER COMPLEX IMAGERY BACKGROUNDS
    Yu, Hang
    Yang, Shihang
    Liu, Zhiheng
    Zhou, Suiping
    IGARSS 2023 - 2023 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM, 2023, : 6406 - 6409