Due to the problems of color distortion and low contrast in the acquired underwater images, an underwater image enhancement algorithm based on unsupervised adaptive uncertainty distribution is proposed in this paper. Firstly, the fog removal module and three traditional enhancement algorithms are used to generate different underwater reference images. Then, a multi-color space stretching module guided by statistics is used to correct the color of the reference image and enhance the contrast of the image. Then the image detail features are obtained by the feature extractor. Finally, Progressive Adaptive Instance Normalization and Laplacian convolution are combined to enhance statistical features, enhance image edge texture and improve image visual effect. Experimental results show that the proposed method can effectively improve image contrast and color deviation, and compared with the existing algorithm, the peak signal-to-noise ratio index of the proposed algorithm is increased by 20.78%, the structural similarity index is increased by 4.23%, and the underwater color image evaluation index is increased by 10.06%.
|