EMO-YOLO: a lightweight ship detection model for SAR images based on YOLOv5s

被引:3
作者
Pan, Hao [1 ]
Guan, Shaopeng [1 ]
Jia, Wanhai [1 ]
机构
[1] Shandong Technol & Business Univ, Sch Informat & Elect Engn, Yantai 264003, Peoples R China
关键词
Synthetic aperture radar; Ship detection; Lightweight model; Feature extraction; DATASET;
D O I
10.1007/s11760-024-03258-2
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Enhancing maritime alert capabilities relies on effective Synthetic Aperture Radar (SAR) ship detection, often achieved through deep learning techniques. However, existing SAR ship detection models face challenges due to their large sizes, rendering them impractical for deployment on resource-constrained devices. Moreover, the complexity of ship backgrounds and the small size of ship targets contribute to decreased detection accuracy. In response, this paper proposes a lightweight ship detection model based on YOLOv5s. Our approach involves restructuring the original model, with a focus on an Efficient Model as the backbone. We achieve model lightweighting by employing a simple stacked Inverted Residual Mobile Block. Additionally, we introduce an enhancement feature extraction module, SCConv_C3, which utilizes Spatial and Channel Reconstruction Convolution (SCConv) to eliminate channel and spatial redundancies in the image while enhancing feature representation capabilities. Furthermore, we integrate Triplet Attention after feature fusion to enhance detection capabilities for small ship targets. Experimental evaluations were conducted on four public datasets. The results demonstrate that our proposed lightweight model maintains high detection accuracy even in challenging scenarios, including complex backgrounds and small ship targets. Notably, on the SSDD dataset, the AP50\documentclass[12pt]{minimal} \usepackage{amsmath} \usepackage{wasysym} \usepackage{amsfonts} \usepackage{amssymb} \usepackage{amsbsy} \usepackage{mathrsfs} \usepackage{upgreek} \setlength{\oddsidemargin}{-69pt} \begin{document}$$_{50}$$\end{document} value reaches 97.8%, surpassing other advanced detection models such as YOLOv5s.
引用
收藏
页码:5609 / 5617
页数:9
相关论文
共 41 条
[1]   Feature Enhancement Pyramid and Shallow Feature Reconstruction Network for SAR Ship Detection [J].
Bai, Lin ;
Yao, Cheng ;
Ye, Zhen ;
Xue, Dongling ;
Lin, Xiangyuan ;
Hui, Meng .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2023, 16 :1042-1056
[2]   Speckle-Free SAR Image Ship Detection [J].
Chen, Si-Wei ;
Cui, Xing-Chao ;
Wang, Xue-Song ;
Xiao, Shun-Ping .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2021, 30 :5969-5983
[3]   Multi-Scale Ship Detection Algorithm Based on YOLOv7 for Complex Scene SAR Images [J].
Chen, Zhuo ;
Liu, Chang ;
Filaretov, V. F. ;
Yukhimets, D. A. .
REMOTE SENSING, 2023, 15 (08)
[4]   A CenterNet plus plus model for ship detection in SAR images [J].
Guo, Haoyuan ;
Yang, Xi ;
Wang, Nannan ;
Gao, Xinbo .
PATTERN RECOGNITION, 2021, 112
[5]  
Huang R, 2018, IEEE INT CONF BIG DA, P2503, DOI 10.1109/BigData.2018.8621865
[6]  
Jocher G., 2022, ZENODO
[7]   Unsupervised change detection in SAR images based on generalized likelihood ratio test and a two-stage morphological filter [J].
Kakoolvand, Abbas ;
Imani, Maryam ;
Ghassemian, Hassan .
INTERNATIONAL JOURNAL OF REMOTE SENSING, 2022, 43 (12) :4630-4662
[8]   SCConv: Spatial and Channel Reconstruction Convolution for Feature Redundancy [J].
Li, Jiafeng ;
Wen, Ying ;
He, Lianghua .
2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR, 2023, :6153-6162
[9]  
Li JW, 2017, C SAR BIG DATA ERA M
[10]   Improved Ship Detection Algorithm Based on YOLOX for SAR Outline Enhancement Image [J].
Li, Sen ;
Fu, Xiongjun ;
Dong, Jian .
REMOTE SENSING, 2022, 14 (16)