Maritime ship detection algorithm based on improved YOLOv4

Wangcheng Chen; Yingshu Li; Xuemei Wang; Mingjing Huang; Zixiang Kang; Xiang Chen

doi:10.1117/12.2660064

3 February 2023 Maritime ship detection algorithm based on improved YOLOv4

Wangcheng Chen, Yingshu Li, Xuemei Wang, Mingjing Huang, Zixiang Kang, Xiang Chen

Proceedings Volume 12511, Third International Conference on Computer Vision and Data Mining (ICCVDM 2022); 125112P (2023) https://doi.org/10.1117/12.2660064
Event: Third International Conference on Computer Vision and Data Mining (ICCVDM 2022), 2022, Hulun Buir, China

Abstract

Maritime ship detection technology has important value in both the military field and maritime supervision. In terms of traditional detection method of maritime ship with low accuracy under complicated situations, in this paper, we adopt a new detection approach based on the improvement of YOLOv4 in order to realize automatic testing of maritime ship under complex circumstances by deep learning. It aims to adopt lightweight network GhostNet as features to extract the network. Depth-separable convolution will be converted to pointwise convolution first and then transformed into depthwise convolution. The network parameter will be reduced while ensuring the accuracy of testing. The accuracy of testing of maritime ship will be further improved by revising activation function as SMU, combining lose function Alpha-IoU and redesigning lose function CIOU. In order to verify the performance of the algorithm in foggy environment, the interference of foggy weather environment is fully considered when generating the training dataset of maritime ships. During training, Mosaic data enhancements were added to the samples to enhance experimental robustness. The loss function was improved using label smoothing techniques to prevent overfitting. Experimental results showed that when the confidence level is 0.5, compared with the original YOLOv4, the average accuracy of the proposed algorithm reaches 99.97% when the number of parameters is reduced by nearly 84.92%. When the ship target is tiny, the testing result is also highly accurate. Therefore, the method can meet the accuracy requirements of real-time processing of maritime vessel detection.

Citation Download Citation

Wangcheng Chen, Yingshu Li, Xuemei Wang, Mingjing Huang, Zixiang Kang, and Xiang Chen "Maritime ship detection algorithm based on improved YOLOv4", Proc. SPIE 12511, Third International Conference on Computer Vision and Data Mining (ICCVDM 2022), 125112P (3 February 2023); https://doi.org/10.1117/12.2660064

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
10 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Convolution

Target detection

Feature extraction

Algorithm development

Image enhancement

Image processing

Network architectures

Keywords/Phrases

Search In:

Publication Years