Skip to content

brucelee206/CycleGAN-TensorFlow

 
 

Repository files navigation

CycleGAN-TensorFlow

An implementation of CycleGan using TensorFlow (work in progess).

Original paper: https://arxiv.org/abs/1703.10593

Environment

  • TensorFlow 1.0.0
  • Python 3.6.0

Data preparing

  • First, download a dataset, e.g. apple2orange
$ bash download_dataset.sh apple2orange
  • Write the dataset to tfrecords
$ python dump.py

Training

$ python train.py

Check TensorBoard to see training progress and generated images.

$ tensorboard --logdir checkpoints/${datetime}
  • apple -> orange
apple2orange apple2orange
  • orange -> apple
orang2apple orang2apple

TODO:

  • Instance normalization
  • Update discriminators using a history of generated images

About

An implementation of CycleGan using TensorFlow.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 96.7%
  • Shell 3.3%