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

Autor: Haochen Wang, Juan Shi, Hamed Karimian, Fucheng Liu, Fei Wang
Jazyk: angličtina
Rok vydání: 2024
Předmět:
Zdroj: International Journal of Digital Earth, Vol 17, Iss 1 (2024)
Druh dokumentu: article
ISSN: 17538947
1753-8955
1753-8947
DOI: 10.1080/17538947.2024.2405525
Popis: 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.
Databáze: Directory of Open Access Journals
Externí odkaz: