Unsupervised haze removal from underwater images

by   Praveen Kandula, et al.

Several supervised networks exist that remove haze information from underwater images using paired datasets and pixel-wise loss functions. However, training these networks requires large amounts of paired data which is cumbersome, complex and time-consuming. Also, directly using adversarial and cycle consistency loss functions for unsupervised learning is inaccurate as the underlying mapping from clean to underwater images is one-to-many, resulting in an inaccurate constraint on the cycle consistency loss. To address these issues, we propose a new method to remove haze from underwater images using unpaired data. Our model disentangles haze and content information from underwater images using a Haze Disentanglement Network (HDN). The disentangled content is used by a restoration network to generate a clean image using adversarial losses. The disentangled haze is then used as a guide for underwater image regeneration resulting in a strong constraint on cycle consistency loss and improved performance gains. Different ablation studies show that the haze and content from underwater images are effectively separated. Exhaustive experiments reveal that accurate cycle consistency constraint and the proposed network architecture play an important role in yielding enhanced results. Experiments on UFO-120, UWNet, UWScenes, and UIEB underwater datasets indicate that the results of our method outperform prior art both visually and quantitatively.


page 1

page 2

page 5


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

Underwater vision suffers from severe effects due to selective attenuati...

Single Underwater Image Restoration by Contrastive Learning

Underwater image restoration attracts significant attention due to its i...

See Blue Sky: Deep Image Dehaze Using Paired and Unpaired Training Images

The issue of image haze removal has attracted wide attention in recent y...

Zero shot framework for satellite image restoration

Satellite images are typically subject to multiple distortions. Differen...

Underwater Image Enhancement Using Pre-trained Transformer

The goal of this work is to apply a denoising image transformer to remov...

Unsupervised Domain-Specific Deblurring via Disentangled Representations

Image deblurring aims to restore the latent sharp images from the corres...

Unsupervised Restoration of Weather-affected Images using Deep Gaussian Process-based CycleGAN

Existing approaches for restoring weather-degraded images follow a fully...

Please sign up or login with your details

Forgot password? Click here to reset