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.

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

Depth-Conditioned GAN for Underwater Image Enhancement

Underwater Attentional Generative Adversarial Networks for Image Enhancement

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

AUIE–GAN: Adaptive Underwater Image Enhancement Based on Generative Adversarial Networks

Underwater Image Enhancement Using a Multiscale Dense Generative Adversarial Network

Generative adversarial networks with multi-scale and attention mechanisms for underwater image enhancement

Hierarchical attention aggregation with multi-resolution feature learning for GAN-based underwater image enhancement

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

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

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

A Novel Generative Adversarial Network for Underwater Image Enhancement

Towards Real-Time Advancement of Underwater Visual Quality With GAN

Underwater image enhancement based on a portion denoising adversarial network

ULD-CycleGAN: An Underwater Light Field and Depth Map-Optimized CycleGAN for Underwater Image Enhancement

MuLA-GAN: Multi-Level Attention GAN for Enhanced Underwater Visibility

Underwater image enhancement based on conditional generative adversarial network

Underwater low-light enhancement network based on bright channel prior and attention mechanism

Underwater single-image restoration based on modified generative adversarial net

Enhancement of Underwater Images by CNN-Based Color Balance and Dehazing

WaterGAN: Unsupervised Generative Network to Enable Real-time Color Correction of Monocular Underwater Images