/CycleGAN

Tensorflow implementation of CycleGAN

Primary LanguagePython

CycleGAN

Tensorflow implementation of CycleGAN.

  1. Original implementation
  2. Paper

Blog

If you would like to understand the paper and see how to implement it by your own, you can have look at the blog by me

Both code and blog received mention on original project homepage over here

CycleGAN model

CycleGAN model can be summarized in the following image. For full details about implementation and understanding CycleGAN you can read the tutorial at this link

Model

Generator

Generator

Discriminator

Discriminator

Our Results

We ran the model for horse2zebra dataset but because of the lack of resources, we just ran the model for 100 epochs and got following results.

Results

Final Comments

  1. During training we noticed that the ouput results were sensitive to initialization. Thanks to vanhuyz for pointing this out and suggesting training multiple times to get best results. You might notice background color being reversed as in following image. This effect can be observed only after 10-20 epochs and you can try to run the code again.

Fail

  1. We also think that this model is not good fit to change the shape of object. We tried to run the model for converting a men's face to a look alike women's face. For that we used celebA dataset but the results are not good and images produced are quite distorted.