Enhanced Multi-Scale Ship Detection Through Joint Multi-Source Image Training
In order to improve the detection performance of traditional single-source remote sensing image ship detection methods in terms of anti-interference capability and multi-scale targets. We develop an improved algorithm based on YOLOv8, called Sea-YOLOv8, and obtain the multi-source remote sensing image detection model SAR-YOLOv8 through migration learning.First, Efficient Multi-Scale Convolution (EMSConv) is used to improve the ability of the C2f module in capturing the features of ships at different scales by using different sizes of convolution kernels. scale ship features using different sized convolution kernels. Secondly, Large Separable Kernel Attentio (LSKAtt) is applied to the SPPF module to decompose the 2D convolution kernel into horizontal and vertical 1D kernels, which improves the computational efficiency while maintaining the high efficiency image processing capability. Experimental results on the SeaShips and SSDD datasets show that Sea-YOLOv8 has a good performance in terms of R, mAP, model parameters and FPS have achieved satisfactory results. Finally, the optimal transfer learning strategy was determined through layer-by-layer freezing, training and validation, which led to the construction of the multi-source remote sensing image recognition model SAR-YOLOv8.
Authors
- Minghua Cao (ORCID: https://orcid.org/0000-0002-1610-2007)
- Yue Zhang (ORCID: https://orcid.org/0000-0001-5318-2475)
- Ning Cheng (ORCID: https://orcid.org/0000-0002-0988-5023)
- Yuan Huang (ORCID: https://orcid.org/0009-0005-1038-1518)
Institutions
- Lanzhou University of Technology (CN)
Publication Details
- Published
- 2024-08-21
- DOI
- https://doi.org/10.1109/icicn62625.2024.10761703
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China