yd8534976 / conGAN

pix2pix-tensorflow

conditional GAN (pix2pix-tensorflow)

[Paper] [Torch ver.]

Setup

Prerequisites

• Linux
• GPU
• tensorflow-1.4
• tensorboard (optional)

Getting Started

• Clone this repo:

git clone https://github.com/yd8534976/conGAN.git
cd conGAN

• Download the dataset:

python ./tools/download-dataset.py facades
mv facades dataset

• Train mode

python main.py --mode="train"

• Test mode

python main.py --mode="test"

Visualization

tensorboard --logdir=summary/

Citation

If you use this code for your research, cite paper Image-to-Image Translation with Conditional Adversarial Networks

@article{pix2pix2016,
  title={Image-to-Image Translation with Conditional Adversarial Networks},
  author={Isola, Phillip and Zhu, Jun-Yan and Zhou, Tinghui and Efros, Alexei A},
  journal={arxiv},
  year={2016}
}