the pytorch version of pix2pix
- CUDA 8.0+
- pytorch 0.3.1
- torchvision
- Download a pix2pix dataset (e.g.facades):
bash ./datasets/download_pix2pix_dataset.sh facadespython pix2pix_train.py --data_root 'your data directory' --which_direction "BtoA"
[1]Image-to-Image Translation with Conditional Adversarial Networks
@inproceedings{isola2017image,
title={Image-to-Image Translation with Conditional Adversarial Networks},
author={Isola, Phillip and Zhu, Jun-Yan and Zhou, Tinghui and Efros, Alexei A},
booktitle={Computer Vision and Pattern Recognition (CVPR), 2017 IEEE Conference on},
year={2017}
}