Underwater GAN (UGAN)

[Project Page] [arxiv] [Video]

Color correction for underwater vision. This project uses CycleGAN to perform style transfer on "clean" underwater photos to appear to be underwater.

We used this video from Youtube as our distorted domain, and hand picked images from Imagenet for our "clean" domain. CycleGAN was used to distort clean images to provide us a dataset.

Name		Name	Last commit message	Last commit date
Latest commit History 266 Commits
cyclegan		cyclegan
datasets		datasets
embedding		embedding
image_interp		image_interp
nets		nets
ops		ops
tests		tests
utils		utils
.gitignore		.gitignore
README.md		README.md
_eval_all.sh		_eval_all.sh
_train.sh		_train.sh
convert_video.py		convert_video.py
eval.py		eval.py
eval_folder.py		eval_folder.py
eval_one.py		eval_one.py
interp_images.py		interp_images.py
train.py		train.py
underwater_eval.py		underwater_eval.py
visualize_filters.py		visualize_filters.py

cameronfabbri/Underwater-Color-Correction

Folders and files

Latest commit

History

Repository files navigation

Underwater GAN (UGAN)

Results

CycleGAN Comparison

About

Resources

Stars

Watchers

Forks

Languages