Underwater target recognition algorithm based on improved convolutional neural network: YOLOV5-Improved

Abstract

This paper proposes an underwater identification method based on the YOLOV5-Improved model. The MobileOne module is added to the Backbone network of the previous model to improve the lightweight model and reduce the network complexity. The amount of parameters, the amount of calculation, and the inference time; the CA full-dimensional dynamic convolution module is embedded in the MP module of the previous model to focus on the main features, and better focus on the discriminative features of fishing nets, thereby improving the accuracy of fishing net recognition; in The adaptive spatial feature fusion module ASFF is introduced into the head network of the previous model to perform multi-scale feature fusion and remove useless background information to improve the detection and mining capabilities of smaller targets in complex backgrounds under natural conditions. The final experiment proved that the YOLOV5-Improved model has a good learning effect and faster convergence speed during the training process, and the experimental accuracy can reach 99.6%, which can better detect underwater fishing nets and provide assistance for safe navigation at sea.