The application discloses an
underwater target recognition method based on an improved YOLOv8
algorithm, which comprises the following steps: inputting an
underwater image into an improved YOLOv8n model for
underwater target detection, and obtaining an output underwater target recognition result. The method has the following advantages: DSConv is used in the
convolution block of the P5 layer of the
backbone network and the last layer of the neck network, which reduces the
network complexity and improves the
inference speed; the C2f_DiRMB module is used in the fourth C2f module of the
backbone network, which introduces an inverted residual attention mechanism and a double-channel
convolution, enhances the ability of the network to capture key
global information, reduces the training parameters, and thus improves the understanding of complex scenes; finally, a
small target detection head is added in the head network to improve the
small target detection capability; and the underwater target recognition method improves the mAP@0.5%, mAP@0.5-0.95%,
precision and recall by 0.5%, 0.8%, 0.5% and 1.0% respectively.