CycleGAN Tensorflow PyTorch
Switch branches/tags
Nothing to show
Clone or download
Latest commit 1cf2c6f May 12, 2018
Permalink
Failed to load latest commit information.
pics add gif2gif May 17, 2017
pytorch ... Apr 11, 2018
LICENSE ... Oct 20, 2017
README.md ... May 12, 2018
data.py new version Apr 13, 2018
download_dataset.sh initial commit May 16, 2017
image_utils.py new version Apr 13, 2018
models.py new version Apr 13, 2018
test.py new version Apr 13, 2018
train.py new version Apr 13, 2018
utils.py new version Apr 13, 2018

README.md

Recommendation


New

  • 2018.04.13: We modify the codes: use the newest tensorflow 1.7 API, and remove the redundancies.

  • 2017.12.22: We add a simple PyTorch implementation, see the "pytorch" folder.

CycleGAN

Tensorflow implementation of CycleGAN.

Paper: Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networks Author: Jun-Yan Zhu et al.

Exemplar results on testset

  • gif: horse -> zebra
  • row 1: horse -> zebra -> reconstructed horse, row 2: zebra -> horse -> reconstructed zebra
  • row 1: apple -> orange -> reconstructed apple, row 2: orange -> apple -> reconstructed orange

Prerequisites

  • tensorflow r1.7
  • python 2.7

Usage

cd CycleGAN-Tensorflow-PyTorch-master

Download Datasets

  • Download the horse2zebra dataset:
sh ./download_dataset.sh horse2zebra
  • Download the apple2orange dataset:
sh ./download_dataset.sh apple2orange
  • See download_dataset.sh for more datasets

Train Example

CUDA_VISIBLE_DEVICES=0 python train.py --dataset=horse2zebra

Test Example

CUDA_VISIBLE_DEVICES=0 python test.py --dataset=horse2zebra