15 February 2024 Lightweight ship detection method based on Swin-YOLOFormer
Jian Cen, Jiahao Chen, Xi Liu, Hao Feng, Jiaxi Li, Haisheng Li, Weisheng Huang
Author Affiliations +
Abstract

Deep learning models have achieved great success in the field of ship detection, but these models often require a large amount of computing and storage resources, and are not suitable for some resource-constrained situations. To solve the above problems, we propose a lightweight Swin-YOLOFormer ship detection method. First, in terms of the backbone network, the Swin transformer lightweight model is introduced to reduce the redundancy parameters of the backbone network. Second, in the feature fusion network, an improved ghost-efficient long-range attention network-hierarchy module is proposed to extract features and reduce the burden of model parameters. Finally, in order to prevent feature loss, an improved SPPCPSC module is proposed to enhance the feature of receptive field. Through experimental verification, compared with the YOLOv7 benchmark model, the number of proposed model parameters was reduced by 66.05% to 13.501 M, which not only accelerated the model training speed but also reached 97.81% accuracy. The results show that the proposed method achieves model lightweight and maintain high precision.

© 2024 SPIE and IS&T
Jian Cen, Jiahao Chen, Xi Liu, Hao Feng, Jiaxi Li, Haisheng Li, and Weisheng Huang "Lightweight ship detection method based on Swin-YOLOFormer," Journal of Electronic Imaging 33(1), 013043 (15 February 2024). https://doi.org/10.1117/1.JEI.33.1.013043
Received: 7 November 2023; Accepted: 31 January 2024; Published: 15 February 2024
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Object detection

Transformers

Feature extraction

Education and training

Performance modeling

Detection and tracking algorithms

Data modeling

Back to Top