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.

A Novel Generative Adversarial Network for Underwater Image Enhancement

Depth-Conditioned GAN for Underwater Image Enhancement

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

Underwater Image Enhancement Using a Multiscale Dense Generative Adversarial Network

Underwater image enhancement based on conditional generative adversarial network

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

Underwater image enhancement based on a portion denoising adversarial network

Underwater Image Enhancement Algorithm Based on Adversarial Training

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

A Novel Underwater Image Enhancement Attention Mechanism Network

Dilated Generative Adversarial Networks for Underwater Image Restoration

Underwater Attentional Generative Adversarial Networks for Image Enhancement

Underwater single-image restoration based on modified generative adversarial net

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

A Multi-Scale Evolutionary Generative Adversarial Network for Underwater Image Enhancement

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

A two-stage and two-branch generative adversarial network-based underwater image enhancement

Enhancing Underwater Imagery Using Generative Adversarial Networks

Towards Real-Time Advancement of Underwater Visual Quality With GAN

A Generative Approach for Detection-driven Underwater Image Enhancement

Global feature fusion generative adversarial network for underwater image enhancement