Skip to content

Using GANs to correct color distortion in underwater images.

Notifications You must be signed in to change notification settings

balzac001/Underwater-Color-Correction

 
 

Repository files navigation

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.

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 97.9%
  • Shell 2.1%