Ship-YOLOv5s: improved YOLOv5 ship target detection based on attention mechanism

Peng Zhang; Ze Sun; Junwei Dong; Jiale Zhang; Yulin Bai

doi:10.1117/12.3045113

16 January 2025 Ship-YOLOv5s: improved YOLOv5 ship target detection based on attention mechanism

Peng Zhang, Ze Sun, Junwei Dong, Jiale Zhang, Yulin Bai

Author Affiliations +

Proceedings Volume 13447, International Conference on Mechatronics and Intelligent Control (ICMIC 2024); 134470S (2025) https://doi.org/10.1117/12.3045113
Event: International Conference on Mechatronics and Intelligent Control (ICMIC 2024), 2024, Wuhan, China

Abstract

The advancement of deep learning technology has greatly improved object detection in a variety of settings. The detection accuracy of multiple ship targets under a complicated background is still poor in the field of ships. In this paper, a SHIP-YOLOV5S model for multi-scale ship target detection and recognition is proposed by improving the YOLOv5 model. Especially the introduction of the Swin Transformer module, enabling the model to capture richer context and global information, and improve the detection performance of multi-scale ship targets. SimAM is simultaneously integrated into the model to enhance the detection accuracy of the model by highlighting the more important feature information and suppressing the less important feature information. So as to improve the detection accuracy of the model. The improved Ship-YOLOv5s model was compared and validated on the collection-produced ship dataset ShipData. The results show that the improved Ship-YOLOv5s model has a detection accuracy of 82.7%, a recall of 77.9%, and a mAP of 80.7% for ship targets, which are 2.1%, 1.0%, and 0.5%, respectively, compared to the YOLOv5 model. It shows that the improved model has more excellent ship target recognition and detection performance, which lays the foundation for subsequent applications in the fields of harbor management, navigation safety, and maritime search and rescue.

(2025) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Peng Zhang, Ze Sun, Junwei Dong, Jiale Zhang, and Yulin Bai "Ship-YOLOv5s: improved YOLOv5 ship target detection based on attention mechanism", Proc. SPIE 13447, International Conference on Mechatronics and Intelligent Control (ICMIC 2024), 134470S (16 January 2025); https://doi.org/10.1117/12.3045113

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

;

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
9 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Target detection

Performance modeling

Target recognition

Transformers

Object detection

Data modeling

Education and training

Show All Keywords

Keywords/Phrases

Search In:

Publication Years