Code for All-In-One Underwater Image Enhancement using Domain-Adversarial Learning [paper]
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