Code for All-In-One Underwater Image Enhancement using Domain-Adversarial Learning [paper] [arXiv]
Synthesized NYU Depth V2 Underwater Dataset based on Anwar et al. (2018)
All the dependencies can be installed by creating a conda environment from the environment.yml file as follows
conda env create --name envname -f=environments.yml
Some parts of code adopted from https://github.com/milesial/Pytorch-UNet and https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix
If you find this work helpful and plan to use this in your project, please cite us by using the following bibtex
@InProceedings{Uplavikar_2019_CVPR_Workshops,
author = {M Uplavikar, Pritish and Wu, Zhenyu and Wang, Zhangyang},
title = {All-in-One Underwater Image Enhancement Using Domain-Adversarial Learning},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR) Workshops},
month = {June},
year = {2019}
}