Abstract:In recent years, with the increasingly serious problems of resource shortage and environmental pollution, the exploration and development of underwater clean energy were particularly important. At the same time, abundant underwater resources and species have attracted a large number of scientists to carry out research on underwater-related tasks. Due to the diversity and complexity of underwater environments, it is difficult to perform related vision tasks, such as underwater target detection and capture. The development of digital image technology has been relatively mature, and it has been applied in many fields and achieved remarkable results, but the research on underwater image processing technology is rarely effective. The underwater environment is much more complicated than that on land, and there is no light source underwater. Underwater imaging systems must rely on artificial light sources for illumination. When light travels through water, it is severely attenuated by water absorption, reflection, and scattering. The collected underwater images inevitably have problems such as limited visible range, blur, low contrast, uneven illumination, incoherent colors, and noise. The purpose of image enhancement is to improve or solve one or more of the above problems in a targeted manner. Therefore, underwater image enhancement technology has become one of the key contents of underwater image processing technology research. In this paper, we proposed a conditional generative adversarial network model based on attention U-Net which contains an attention gate mechanism that could filter invalid feature information and capture contour, local texture, and style information effectively. Furthermore, we formulate an objective function through three different loss functions, which can evaluate image quality from global content, color, and structural information. Finally, we performed end-to-end training on the UIEB real-world underwater image dataset. The comparison experiments show that our method outperforms all comparative methods, the ablation experiments show that the loss function proposed in this paper outperforms a single loss function, and finally, the generalizability of our method is verified by executing on two different datasets, UIEB and EUVP.

Underwater Image Enhancement based on Retinex Decomposition and Unsupervised Generative Adversarial Networks

Depth-Conditioned GAN for Underwater Image Enhancement

Underwater single-image restoration based on modified generative adversarial net

Real-World Underwater Image Enhancement Based on Attention U-Net

Underwater Image Enhancement Using a Multiscale Dense Generative Adversarial Network

Underwater image enhancement based on a portion denoising adversarial network

Underwater Attentional Generative Adversarial Networks for Image Enhancement

Underwater image enhancement by combining multi-attention with recurrent residual convolutional U-Net

Two-Stage Underwater Image Restoration Algorithm Based on Physical Model and Causal Intervention

UWGAN: Underwater GAN for Real-world Underwater Color Restoration and Dehazing

A Novel Generative Adversarial Network for Underwater Image Enhancement

Underwater Image Enhancement Algorithm Based on Adversarial Training

Deep retinex decomposition network for underwater image enhancement

An Underwater Image Enhancement Method for a Preprocessing Framework Based on Generative Adversarial Network

Underwater image enhancement based on conditional generative adversarial network

RAUNE-Net: A Residual and Attention-Driven Underwater Image Enhancement Method

An optimized GAN method based on the Que-Attn and contrastive learning for underwater image enhancement

Enhanced visual perception for underwater images based on multistage generative adversarial network

A Novel Underwater Image Enhancement Attention Mechanism Network

A deep Retinex network for underwater low-light image enhancement

UP‐GAN: Channel‐spatial attention‐based progressive generative adversarial network for underwater image enhancement