Marine Target Detection Algorithm Based on Deep Learning

Yuchao Wang,Yong Zhou,Xiaojing Xia
DOI: https://doi.org/10.1109/cac53003.2021.9727659
2021-01-01
Abstract:Aiming at the problems of low detection accuracy and slow training speed of YOLOv3 for marine target, a ship target detection method based on the improved YOLOv3 algorithm is proposed. In this method, the BN layer is merged into the convolutional layer to improve the forward inference speed of the model. GloU is selected to replace the original mean square error loss function, which can not only focus on overlapping areas, but also other non-overlapping areas to improve the detection accuracy. In order to solve the problem of image distortion, blur and low contrast, gaussian filter brightness enhancement and histogram equalization were used to preprocess the image to improve the image quality. The preprocessed ship image was input into the improved YOLOv3 network for training. The experimental results show that, compared with the original YOLOv3 algorithm, the training speed is improved by 7.69%, the reasoning speed is improved by 27.47%, and the AP is improved by 3.12%, which verifies the effectiveness of the proposed algorithm in marine ship target detection.
What problem does this paper attempt to address?