Automation & Computer Technologies
Wang Yan, Wang Likang, Zhang Jinfeng, Fan Xianghui
In recent years, underwater image enhancement techniques has received a wide range of attention from related researchers with the rise of marine resource exploitation. As the existing network feature extraction is not sufficient and the enhancement results have the problems of incomplete defogging and inaccurate color bias correction, in this paper, an underwater image enhancement method based on global dense two-branch cascade network and spatial domain grayscale transformation is proposed. The global dense two-branch cascade network can amplify the global dimensional interaction features while reducing information reduction on the one hand, and extract spatial features by obtaining spatial information at different scales to achieve richer feature extraction on the other hand; the spatial domain grayscale transformation operation can improve the contrast while color correcting the image, which makes the image visual effect better. After the training is completed, an end-to-end inference can be performed on the underwater images. The experimental results show that this paper’s model works best on the EUVP dataset, and compared with the second best, this paper’s model obtains 3.371, 0.06, 0.716, 0.024, and 1.727 improvements in PSNR, SSIM, UIQM, UCIQE, and CCF, respectively. Compared with other representative methods, the proposed network achieves significant visual enhancement in dealing with severe color bias, low light, and detail loss in underwater images.