Abstract:Images captured in underwater environments often suffer from color distortion, low contrast, and reduced visual quality. Most existing methods solve underwater image enhancement (UIE) by applying supervised training on synthetic images or pseudo references. However, the synthetic paired data fail to accurately replicate real-world data due to the inherent differences, and the quantity and quality of pseudo references are limited, which seriously reduces the generalization ability and performance of the model when testing on real underwater images. In contrast, unsupervised-based method is not constrained by paired data, which is more robust and potentially more promising for practical applications. Nevertheless, existing unsupervised-based methods cannot effectively constrain the network to train a model that can adapt to various degradation. Inspired by the fact that people often resolve problems from opposing but complementary perspectives, we maintain that there is implicit cooperation between the removal and generation of water layers, as they can constrain and promote each other at the same time. Based on the above analysis, a new unsupervised-based UIE framework that jointly learns water layer generation and removal based on disentangled representations is proposed. Specifically, we propose a bidirectional disentangling network in which each unidirectional network contains a loop consisting of water layer removal and generation, and restricts the image to remain consistent after a loop. Meanwhile, a novel double-order contrastive loss is proposed to improve the ability of disentanglement by utilizing the joint implicit constraint of first-order features and second-order features. Extensive experimental results demonstrate that the model outperforms the state-of-the-art methods in both qualitative and quantitative evaluation with a relatively high processing speed. The experimental results of the ablation study demonstrate the usefulness of the various components.

Dual Adversarial Contrastive Learning for Underwater Image Enhancement

Twin Adversarial Contrastive Learning for Underwater Image Enhancement and Beyond

Deformable Convolutional Network Constrained by Contrastive Learning for Underwater Image Enhancement

Dual-branch Contrastive Learning for Image Enhancement of Underwater Internet of Things

Underwater Image Enhancement Algorithm Based on Adversarial Training

Unsupervised Underwater Image Enhancement Based on Disentangled Representations Via Double-Order Contrastive Loss

Underwater Image Enhancement Network Based on Dual Layers Regression

Fast Adaptive Self-Supervised Underwater Image Enhancement.

HCLR-Net: Hybrid Contrastive Learning Regularization with Locally Randomized Perturbation for Underwater Image Enhancement

Underwater Image Enhancement with Cascaded Contrastive Learning

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

Hybrid CNN-Transformer Network for Two-Stage Underwater Image Enhancement with Contrastive Learning

Semi-supervised advancement of underwater visual quality

All-In-One Underwater Image Enhancement using Domain-Adversarial Learning

Attention-based color consistency underwater image enhancement network

Underwater Image Enhancement Based on Improved U-Net Convolutional Neural Network

Underwater Image Enhancement Based on Color Restoration and Dual Image Wavelet Fusion

Emerging from Water: Underwater Image Color Correction Based on Weakly Supervised Color Transfer

Underwater Image Enhancement Via Piecewise Color Correction and Dual Prior Optimized Contrast Enhancement.

Dual Adversarial Resilience for Collaborating Robust Underwater Image Enhancement and Perception.

Underwater Image Restoration via Contrastive Learning and a Real-World Dataset