Dream-Runner-Yu / Underwater-Color-Correction

Using GANs to correct color distortion in underwater images.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Underwater GAN (UGAN)

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

About

Using GANs to correct color distortion in underwater images.


Languages

Language:Python 97.9%Language:Shell 2.1%