An underwater target recognition method based on improved YOLOv4 in complex marine environment

Abstract

In the marine environment, there are problems such as complex background and low illumination, resulting in poor picture quality, and the aggregation of small targets and multiple targets brings difficulties to target recognition. In order to improve the accuracy of marine target detection, the image enhancement and improved YOLOv4 algorithm are used to identify marine organisms. Firstly, aiming at the problems of image blur, some of the images are enhanced with the multi-scale retinex with colour restoration (MSRCR) enhancement algorithm so that the image is clearer and it is easier to extract features. Secondly, Mosaic augmentation is added to YOLOv4 to enrich the data set and increase network robustness. Then the Spatial Pyramid Pooling (SPP) module of YOLOv4 is improved by changing the size of the pooling core, which increases the range of feature extraction and improves the detection capabilities, and its mAP value reaches 97.06%. The experimental results show that the detection accuracy of the image enhancement and improved YOLOv4 algorithm is 7.16% higher than that of the original algorithm. And the improved YOLOv4 algorithm is improved in average precision and recall rate compared with other algorithms, which verifies the effectiveness of the algorithm.