Skip to content

yd8534976/conGAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

conditional GAN (pix2pix-tensorflow)

[Paper] [Torch ver.]

examples

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}
}