Skip to content
CycleGAN Tensorflow 2
Python Shell
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
imlib
pics
pylib Tensorflow 2.0 Alpha Apr 6, 2019
tf2gan Tensorflow 2.0 Alpha Apr 6, 2019
tf2lib Tensorflow 2.0 Alpha Apr 6, 2019
.gitignore Tensorflow 2.0 Alpha Apr 6, 2019
LICENSE
README.md update README.md Jun 21, 2019
data.py Tensorflow 2.0 Alpha Apr 6, 2019
download_dataset.sh initial commit May 16, 2017
module.py
test.py
train.py

README.md

Recommendation

  • Our GAN based work for facial attribute editing - AttGAN.

New

  • We re-implement CycleGAN by Tensorflow 2! The old versions are here: v1, v0.


CycleGAN - Tensorflow 2

Tensorflow 2 implementation of CycleGAN.

Paper: Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networks

Author: Jun-Yan Zhu et al.

Exemplar results

summer2winter

row 1: summer -> winter -> reconstructed summer, row 2: winter -> summer -> reconstructed winter

horse2zebra

row 1: horse -> zebra -> reconstructed horse, row 2: zebra -> horse -> reconstructed zebra

apple2orange

row 1: apple -> orange -> reconstructed apple, row 2: orange -> apple -> reconstructed orange

Usage

  • Prerequisites

    • Tensorflow 2.0 Alpha pip install tensorflow-gpu==2.0.0-alpha0
    • Tensorflow Addons pip install tensorflow-addons
    • (if you meet "tf.summary.histogram fails with TypeError" pip install --upgrade tb-nightly)
    • scikit-image, oyaml, tqdm
    • Python 3.6
  • Dataset

    • download the summer2winter dataset

      sh ./download_dataset.sh summer2winter_yosemite
    • download the horse2zebra dataset

      sh ./download_dataset.sh horse2zebra
    • see download_dataset.sh for more datasets

  • Example of training

    CUDA_VISIBLE_DEVICES=0 python train.py --dataset summer2winter_yosemite
    • tensorboard for loss visualization

      tensorboard --logdir ./output/summer2winter_yosemite/summaries --port 6006
  • Example of testing

    CUDA_VISIBLE_DEVICES=0 python test.py --experiment_dir ./output/summer2winter_yosemite
You can’t perform that action at this time.