• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

Project: CycleGAN-Tensorflow-2 (GitHub Link)

• CycleGAN-Tensorflow-2-master
  • module.py
  • data.py
  • tf2gan
    • loss.py
    • __init__.py
  • imlib
    • basic.py
    • transform.py
    • dtype.py
    • __init__.py
  • download_dataset.sh
  • LICENSE
  • pylib
    • path.py
    • argument.py
    • processing.py
    • serialization.py
    • __init__.py
    • timer.py
  • tf2lib
    • utils
      • __init__.py
      • utils.py
    • data
      • __init__.py
      • dataset.py
    • __init__.py
    • ops
      • ops.py
      • __init__.py
    • image
      • image.py
      • __init__.py
  • pics
  • train.py
  • README.md
  • test.py
  • .gitignore

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 pip install tensorflow-gpu
  • 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