Abstract:Underwater image enhancement (UIE) is a highly challenging task due to the complexity of underwater environment and the diversity of underwater image degradation. Due to the application of deep learning, current UIE methods have made significant progress. Most of the existing deep learning-based UIE methods follow a single-stage network which cannot effectively address the diverse degradations simultaneously. In this paper, we propose to address this issue by designing a two-stage deep learning framework and taking advantage of cascaded contrastive learning to guide the network training of each stage. The proposed method is called CCL-Net in short. Specifically, the proposed CCL-Net involves two cascaded stages, i.e., a color correction stage tailored to the color deviation issue and a haze removal stage tailored to improve the visibility and contrast of underwater images. To guarantee the underwater image can be progressively enhanced, we also apply contrastive loss as an additional constraint to guide the training of each stage. In the first stage, the raw underwater images are used as negative samples for building the first contrastive loss, ensuring the enhanced results of the first color correction stage are better than the original inputs. While in the second stage, the enhanced results rather than the raw underwater images of the first color correction stage are used as the negative samples for building the second contrastive loss, thus ensuring the final enhanced results of the second haze removal stage are better than the intermediate color corrected results. Extensive experiments on multiple benchmark datasets demonstrate that our CCL-Net can achieve superior performance compared to many state-of-the-art methods. The source code of CCL-Net will be released at <a class="link-external link-https" href="https://github.com/lewis081/CCL-Net" rel="external noopener nofollow">this https URL</a>.

CFENet: Cost-effective Underwater Image Enhancement Network Via Cascaded Feature Extraction

A Multi-Scale Feature Extraction Network for Marine Underwater Image Enhancement

An Efficient Network for Underwater Image Enhancement with Gated Linear Units and Fast Fourier Convolution

Robust underwater image enhancement with cascaded multi-level sub-networks and triple attention mechanism

LiteEnhanceNet: A Lightweight Network for Real-Time Single Underwater Image Enhancement

Underwater Image Super-Resolution and Enhancement Via Progressive Frequency-Interleaved Network

PDCFNet: Enhancing Underwater Images through Pixel Difference Convolution

FDCE-Net: Underwater Image Enhancement with Embedding Frequency and Dual Color Encoder

A Deep Cnn Method For Underwater Image Enhancement

A Cascaded Multimodule Image Enhancement Framework for Underwater Visual Perception

Adaptive Deep Learning Network with Multi-Scale and Multi-Dimensional Features for Underwater Image Enhancement

A Novel Underwater Image Enhancement Using Optimal Composite Backbone Network

UDA‐Net: Densely Attention Network for Underwater Image Enhancement

Underwater Image Enhancement Network Based on Dual Layers Regression

Three-channel Cascade Network for Underwater Image Enhancement

Underwater Image Enhancement with Cascaded Contrastive Learning

Underwater image enhancement with encoding-decoding deep CNN networks

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

Underwater Image Enhancement Via Cross-Wise Transformer Network Focusing on Pre-Post Differences

Underwater Image Enhancement Based on Dual U-net

LAFFNet: A Lightweight Adaptive Feature Fusion Network for Underwater Image Enhancement