PyTorch implementation of Image-to-Image Translation Using Conditional Adversarial Networks.
Based on pix2pix by Phillip Isola et al.
The examples from the paper:
- Linux
- Python, Numpy, PIL
- pytorch 0.4.0
- torchvision 0.2.1
-
Clone this repo:
git clone [email protected]:mrzhu-cool/pix2pix-pytorch.git cd pix2pix-pytorch
-
Get dataset
unzip dataset/facades.zip
-
Train the model:
python train.py --dataset facades --cuda
-
Test the model:
python test.py --dataset facades --cuda
This code is a simple implementation of pix2pix. Easier to understand. Note that we use a downsampling-resblocks-upsampling structure instead of the unet structure in this code, therefore the results of this code may inconsistent with the results presented in the paper.
Highly recommend the more sophisticated and organized code pytorch-CycleGAN-and-pix2pix by Jun-Yan Zhu.