YOLOv5 ship: An improved ship target detection algorithm study
Authors: 
Wang Pu, Shenhua Yang, Guoquan Chen, Minjie Zheng, Yongfeng Suo, Tao WangAuthors Info & Claims
Wang Pu, Shenhua Yang, Guoquan Chen, Minjie Zheng, Yongfeng Suo, Tao WangAuthors Info & ClaimsPages 106 - 116
Abstract
Aiming at the problem of difficult detection of maritime ship targets, which is prone to false detection, missed detection, and low recognition accuracy. In this paper, an improved ship target detection algorithm, YOLOv5, is proposed. firstly, the attention mechanism CBAM module is added to feature enhancement of image targets, and irrelevant interference information is suppressed based on retaining key features; secondly, the fusion structure of image feature extraction is improved, and the BiFPN structure is used instead of FPN+PAN structure, which realizes bi-directional multi-scale feature fusion and reduces target information loss and missed detection rate; finally, the loss function is optimized, and WIoU v3 with position and size focus factors is used to optimize the loss function, and the relative position and size of the prediction frame and the rear frame are learned more accurately, which effectively enhances the detection capability of small target ships. The experimental results show that the improved algorithm improves the target detection efficiency, the detection accuracy mAP50 value reaches 93.5%, and the FPS (Frames Per Second) goes to 51 frames per second, which can better meet the requirements of real-time detection and improve the target detection ability of ships in different environments with better robustness.
References
[1]
Li H, Deng L, Yang C, Enhanced YOLO v3 Tiny Network for Real-time Ship Detection from Visual Image[J]. IEEE Access, 2021, PP(99):1-1. https://doi.org/10.1109/ACCESS.2021.3053956.
[2]
Huang, Z.; Jiang, X.; Wu, F.; Fu, Y.; Zhang, Y.; Fu, T.; Pei, J. An Improved Method for Ship Target Detection Based on YOLOv4. Appl. Sci. 2023, 13, 1302. https://doi.org/10.3390/app13031302.
[3]
Qi, L.; Li, B.; Chen, L.; Wang, W.; Dong, L.; Jia, X.; Huang, J.; Ge, C.; Xue, G.; Wang, D. Ship Target Detection Algorithm Based on Improved Faster R-CNN. Electronics 2019, 8, 959. https://doi.org/10.3390/electronics8090959.
[4]
Nie, X.; Duan, M.; Ding, Hu, B.; Wong, E.K. Attention Mask R-CNN for Ship Detection and Segmentation From Remote Sensing Images. in IEEE Access. 2020, 8, 199162-199172, https://doi.org/10.1109/ACCESS.2020.2964540.
[5]
Zou, Y.; Zhao, L.; Qin, S.; Pan, M.; Li, Z. Ship target detection and identification based on SSD_MobilenetV2, 2020 IEEE 5th Information Technology and Mechatronics Engineering Conference (ITOEC), Chongqing, China, 2020, pp. 1676-1680, https://doi.org/10.1109/ITOEC
[6]
49072.2020.9141734.
[7]
Chen, D.; Sun, S.; Lei, Z.; Shao, H.; Wang, Y. Ship Target Detection Algorithm Based on Improved YOLOv3 for Maritime Image, Journal of Advanced Transportation, vol. 2021, Article ID 9440212, 11 pages, 2021. https://doi.org/10.1155/2021/9440212.
[8]
Liu, Y.; Zhao, T.; Shen, Z. Small Target Detection for UAV Aerial Images Based on Improved YOLOv3, 2021 4th International Conference on Robotics, Control and Automation Engineering (RCAE), Wuhan, China, 2021, pp. 12-16, https://doi.org/10.1109/RCAE53607.2021.96
[9]
38796.
[10]
Shi, J.; Yang, J.; Zhang, Y. Research on Steel Surface Defect Detection Based on YOLOv5 with Attention Mechanism. Electronics 2022, 11, 3735. https://doi.org/10.3390/electronics11223735.
[11]
Ye, Y.; Zhen, R.; Shao, Z.; Pan, J.; Lin, Y. A Novel Intelligent Ship Detection Method Based on Attention Mechanism Feature Enhancement. J. Mar. Sci. Eng. 2023, 11, 625. https://doi.org/10.3390/jmse11030625.
[12]
Liu, T.; Pang, B.; Zhang, L.; Yang, W.; Sun, X. Sea Surface Object Detection Algorithm Based on YOLO v4 Fused with Reverse Depthwise Separable Convolution (RDSC) for USV. J. Mar. Sci. Eng. 2021, 9, 753. https://doi.org/10.3390/jmse9070753.
[13]
Hong, W.; Ma, Z.; Ye, B.; Yu, G.; Tang, T.; Zheng, M. Detection of Green Asparagus in Complex Environments Based on the Improved YOLOv5 Algorithm. Sensors 2023, 23, 1562. https://doi.org/10.3390/s23031562.
[14]
Zheng, J.; Sun, S.; Zhao, S. Fast ship detection based on lightweight YOLOv5 network. In Proceedings of the 2021 International Conference on Computer Network, Electronic and Automation (ICCNEA), 18-20 June 2021, Chengdu, China (pp. 135-140). IEEE. https://doi.org/10.1049/ipr2.12432.
[15]
Zhou, J.; Jiang, P.; Zou, A.; Chen, X.; Hu, W. Ship Target Detection Algorithm Based on Improved YOLOv5. J. Mar. Sci. Eng. 2021, 9, 908. https://doi.org/10.3390/jmse9080908.
[16]
Shao, Z.; Wu, W.; Wang, Z.; Du, W.; Li, C. SeaShips: A Large-Scale Precisely Annotated Dataset for Ship Detection. in IEEE Transactions on Multimedia. 2018. 20(10), 2593-2604. https://doi.org/10.1109/TMM.2018.2865686.
[17]
Lei, F.; Tang, F.; Li, S. Underwater Target Detection Algorithm Based on Improved YOLOv5. J. Mar. Sci. Eng. 2022, 10, 310. https://doi.org/10.3390/jmse10030310.
[18]
Fang, Y., Ma, Y., Zhang, X. Enhanced YOLOv5 algorithm for helmet wearing detection via combining bi-directional feature pyramid, attention mechanism and transfer learning. Multimed Tools Appl (2023). https://doi.org/10.1007/s11042-023-14395-0.
[19]
Fan, Y.; Li, Y.; Shi, Y.; Wang, S. Application of YOLOv5 Neural Network Based on Improved Attention Mechanism in Recognition of Thangka Image Defects. in Proceedings of the 2021 International Conference on Artificial Intelligence and Information Engineering (AIIE 2021), 2021, pp. 143-147. http://doi.org/10.3837/tiis.2022.01.014.
[20]
Woo, S.; Park, J.; Lee, J. Y.; Kweon, I. S. CBAM: Convolutional Block Attention Module. Proceedings of the European Conference on Computer Vision (ECCV). 2018, pp. 3-19. htps://doi.org/10.1007/978-3-030-01234-2_ 1.
[21]
Xue, Z.; Xu, R.; Bai, D.; Lin, H. YOLO-Tea: A Tea Disease Detection Model Improved by YOLOv5. Forests 2023, 14, 415. https://doi.org/10.3390/f14020415.
[22]
Yao, J.; Fan, X.; Li, B.; Qin, W. Adverse Weather Target Detection Algorithm Based on Adaptive Color Levels and Improved YOLOv5. Sensors 2022, 22, 8577. https://doi.org/10.3390/s22218577.
[23]
Zhang Y, Huang J, Cai F. On Bridge Surface Crack Detection Based on an Improved YOLO v3 Algorithm[J]. IFAC-Papers Online, 2020, 53(2):8205-8210. https://doi.org/10.1016/j.ifacol.2020.12.1994.
[24]
Hao, W.; Han, W.; Han, M.; Li, F. A Novel Improved YOLOv3-SC Model for Individual Pig Detection. Sensors 2022, 22, 8792. https://doi
[25]
.org/10.3390/s22228792.
[26]
Dong, X.; Yan, S.; Duan, C. A lightweight vehicles detection network model based on yolov5. Engineering Applications of Artificial Intelligence: The International Journal of Intelligent Real-Time Automation. 2022, (113-), 113. https://doi.org/10.1016/j.engappai.2022.10
[27]
4914.
[28]
Wang, X.; Zhang, X.; Zhou, N. Improved YOLOv5 with BiFPN on PCB Defect Detection. In Proceedings of the 2nd International Conference on Artificial Intelligence and Computer Engineering (ICAICE), Hangzhou, China, 2021 (pp.196-199). https://doi.org/10.1109/
[29]
ICAICE54393.2021.00045.
[30]
Chen, Z.; Li, X.; Wang, L.; Shi, Y.; Sun, Z.; Sun, W. An Object Detection and Localization Method Based on Improved YOLOv5 for the Teleoperated Robot. Appl. Sci. 2022, 12, 11441. https://doi.org/10.3390/app122211441.
[31]
Cho, Y. J. Weighted Intersection over Union (WIoU): A New Evaluation Metric for Image Segmentation. IEEE Transactions on Pattern Analysis and Machine Intelligence. 2021, 43(2), 712-723. https://doi.org/10.48550/arXiv.2107.09858.
[32]
Tong, Z.; Chen, Y.; Xu, Z.; Yu, R. Wise-IoU: Bounding Box Regression Loss with Dynamic Focusing Mechanism. in IEEE Transactions on Image Processing. 2020, vol. 29, pp. 8276-8285. https://doi.org/10.48550/arXiv.2301.10051.
[33]
Tang, L.; Xie, T.; Yang, Y.; Wang, H. Classroom Behavior Detection Based on Improved YOLOv5 Algorithm Combining Multi-Scale Feature Fusion and Attention Mechanism. Appl. Sci. 2022, 12, 6790. https://doi.org/10.3390/app12136790.
[34]
Du, F.-J.; Jiao, S.-J. Improvement of Lightweight Convolutional Neural Network Model Based on YOLO Algorithm and Its Research in Pavement Defect Detection. Sensors 2022, 22, 3537. https://doi.org/10.3390/s22093537.
[35]
Wang, Q.; Chen, W.; Tang, H.; Pan, X.; Zhao, H.; Yang, B.; Zhang, H.; Gu, W. Simultaneous extracting area and quantity of agricultural greenhouses in large scale with deep learning method and high-resolution remote sensing images. Science of The Total Environment. 2023, Volume 872,2023,162229. ISSN 0048-9697.https://doi.org/10.1016/j.scitotenv.2023.162229.
Index Terms
- YOLOv5 ship: An improved ship target detection algorithm study
Recommendations
Ship-YOLOv5:Ship Target Detection Based on Enhanced Feature Fusion
CNIOT '24: Proceedings of the 2024 5th International Conference on Computing, Networks and Internet of ThingsDue to multiple factors such as bad weather, weak targets of ships, and complex navigational environment, which lead to the problem of low accuracy of the existing ship detection methods. In this paper, we propose a ship target detection method based on ...
Ship Target Detection Algorithm Based on Improved YOLOX_s
2022 IEEE International Conference on Mechatronics and Automation (ICMA)As an important part of ship intelligent navigation, improving the speed and accuracy of target detection can ensure the safe navigation of ships. Aiming at the low accuracy of ship target detection, an improved YOLOX_s network ship target ...
Infrared Sea Surface Ship Target Detection Algorithm Based on Improved YOLOV5
ICCPR '22: Proceedings of the 2022 11th International Conference on Computing and Pattern RecognitionShip target detection on the sea surface is one of the most common scenes in the field of target detection. The accuracy of target detection plays an important role in the field of sea rescue and defense. However, complex infrared sea surface scenes, ...
Comments
Information & Contributors
Information
Published In
October 2023
151 pages
ISBN:9798400708985
DOI:10.1145/3633598
Copyright © 2023 ACM.
Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than the author(s) must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected].
Publisher
Association for Computing Machinery
New York, NY, United States
Publication History
Published: 22 January 2024
Check for updates
Author Tags
Qualifiers
- Research-article
- Research
- Refereed limited
Funding Sources
- National Natural Science Foundation of China
- Naural Science Project of Fujian Province
- Key Projects of the National Key R & D Program
- Fuzhou-Xiamen-Quanzhou Independent Innovation Region Cooperated Special Foundation
Conference
ICAAI 2023
ICAAI 2023: 2023 the 7th International Conference on Advances in Artificial Intelligence
October 13 - 15, 2023
Istanbul, Turkiye
Contributors
Other Metrics
Bibliometrics & Citations
Bibliometrics
Article Metrics
- 0Total Citations
- 50Total Downloads
- Downloads (Last 12 months)43
- Downloads (Last 6 weeks)9
Reflects downloads up to 08 Mar 2025
Other Metrics
Citations
View Options
Login options
Check if you have access through your login credentials or your institution to get full access on this article.
Sign inFull Access
View options
View or Download as a PDF file.
PDFeReader
View online with eReader.
eReaderHTML Format
View this article in HTML Format.
HTML Format