Pix2pix

Yet another TensorFlow implementation of pix2pix

Original paper: Image-to-Image Translation with Conditional Adversarial Networks

Requirements

tensorflow_gpu==1.2.1
numpy==1.14.0
Pillow==5.0.0
imageio==2.2.0

Usages

Download dataset

bash ./download_dataset.sh dataset_name

dataset_name := [facades | cityscapes | maps | edges2shoes | edges2handbags]

Train

python main.py --phase train  --dataset_name facades

Note: Training model will generate a directory like run-xxxxxx-xxxxxx, which contains

• checkpoint/: save model checkpoint
• summaries/: save tensorflow summaries
• log.txt: save training log

Test

python main.py --phase test  --dataset_name facades --prefix run-xxxxxxxx-xxxxxx

and the test results will be displayed in the run-xxxxxxxx-xxxxxx/test/ directory

Result

facades

input	output	target

cityscapes

input	output	target