Skip to content

alecGraves/cyclegan_keras

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

38 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

cyclegan_keras

CycleGAN reimplimented in Keras

What has been made?

  • Instance Norm layer for keras
  • Cycle loss function for keras
  • Test cyclegan training script

What needs to be done?

  • Bigger models for non-mnist images need to be created
  • All of the generator training models should probably be convered into one model for a real problem.
  • Fix the upscaling on generator models

Pictures!

For the test training script, I decided to have cyclegan learn a mapping from cat pictures to mnist pictures (and back). This took like 10 hours on a Nvidia GTX 1060 using the training parameters in mnist_test.py.

Here are some results (from various stages of training):

19

31

34

42

62

72

80

90

Releases

No releases published

Packages

No packages published

Languages