Underwater Distortion Target Recognition Network (UDTRNet) via Enhanced Image Features

Abstract

It is difficult for the autonomous underwater vehicle (AUV) to recognize targets similar to the environment in lacking data labels. Moreover, the complex underwater environment and the refraction of light cause the AUV to be unable to extract the complete significant features of the target. In response to the above problems, this paper proposes an underwater distortion target recognition network (UDTRNet) that can enhance image features. Firstly, this paper extracts the significant features of the image by minimizing the info noise contrastive estimation (InfoNCE) loss. Secondly, this paper constructs the dynamic correlation matrix to capture the spatial semantic relationship of the target and uses the matrix to extract spatial semantic features. Finally, this paper fuses the significant features and spatial semantic features of the target and trains the target recognition model through cross-entropy loss. The experimental results show that the mean average precision (mAP) of the algorithm in this paper increases by 1.52% in recognizing underwater blurred images.