Generative Adversarial Network with Lightweight U-Net for Underwater Optical Image Enhancement

Xinru Wang,Hui Zhou,Xingni Le,Jiarui Ding,Hechuan Tao,Liyan Cheng,Wanru Chen,Ruizhi Wang,Qunhui Yang,Chen,Meiwei Kong
DOI: https://doi.org/10.1109/icwoc62055.2024.10684950
2024-01-01
Abstract:High-quality underwater optical images play a crucial role in marine resources exploration. However, the absorption and scattering of light by seawater can cause underwater optical images to appear blue-green and blurry. Therefore, we propose a generative adversarial network (GAN) with lightweight U-Net for underwater image enhancement, called LUUW-GAN. The adversarial structure of GAN allows the generation of vivid and realistic images, thus better solving the fading problem of underwater images. Moreover, we design the generator of GAN using the lightweight U-Net to preserve fine details of the original input images, thus effectively improving blurring caused by scattering. We incorporate the convolutional block attention module into the skip connections to enhance the feature representation capability. Furthermore, we introduce the top-k selection operator to facilitate self-attention and feature weighting to improve feature aggregation. The experimental results on the enhancement of underwater visual perception dataset show that the proposed method improves the peak signal-to-noise ratio by 1.24 and the structural similarity by 0.021 compared to the basic model.
What problem does this paper attempt to address?