cameronfabbri / Underwater-Color-Correction

Using GANs to correct color distortion in underwater images.
292 stars 86 forks source link

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.

Results

result1

CycleGAN Comparison

result2

Other videos in consideration:

https://www.youtube.com/watch?v=OSdrb1XNXZI

https://www.youtube.com/watch?v=ALN6y2PLCq0