Abstract:In recent years, underwater image enhancement and restoration technologies have become increasingly important in order to optimize the efficiency of maritime operations and promote the automatic machine learning of the maritime industry. A new hierarchical attention aggregation with multi-resolution feature learning for GAN-based underwater image enhancement is proposed to address the problems of color bias, underexposure , and blurring in underwater images. The proposed method consists of a generator and a discriminator. Specifically, the generator includes an encoder, a bottleneck layer, and a decoder. Generator introduces inter-block serial connections for better adaptation to complex image scenes and task requirements, and parallel connections to extract multi-level features and enhance the expressive capacity of the network. To extract semantic and contextual information, hierarchical attention dense aggregation is designed in the encoder, which includes multi-scale feature hierarchy and dense feature hierarchy. Additionally, a multi-scale spatial attention mechanism is designed in the bottleneck layer to handle variations in underwater image scenes. In the decoder, the feature channel layer is emphasized, and a multi-channel attention mechanism is proposed to restore the multi-resolution channel features to a three-channel enhanced image. Furthermore, the angular loss function is introduced as additional supervision, which improves the similarity between the generated and original images, increases image clarity, and reduces color bias. Meanwhile, we employ the patch discriminator to enhance machine vision. Extensive experiments demonstrate that the proposed method outperforms state-of-the-art methods.

Distorted underwater image reconstruction for an autonomous underwater vehicle based on a self-attention generative adversarial network

Depth-Conditioned GAN for Underwater Image Enhancement

Enhancing Underwater Imagery Using Generative Adversarial Networks

Underwater Image Enhancement Using a Multiscale Dense Generative Adversarial Network

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

Underwater single-image restoration based on modified generative adversarial net

Dilated Generative Adversarial Networks for Underwater Image Restoration

Towards Real-Time Advancement of Underwater Visual Quality With GAN

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

A Novel Generative Adversarial Network for Underwater Image Enhancement

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

DAE-GAN: Underwater Image Super-Resolution Based on Symmetric Degradation Attention Enhanced Generative Adversarial Network

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

Underwater Attentional Generative Adversarial Networks for Image Enhancement

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

Real-time Image Enhancement for Vision-based Autonomous Underwater Vehicle Navigation in Murky Waters

Underwater image enhancement based on a portion denoising adversarial network

Two-stage underwater image restoration based on gan and optical model

Underwater image enhancement based on conditional generative adversarial network

Underwater Image Enhancement Algorithm Based on Adversarial Training

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