pix2pix-pytorch
PyTorch implementation of Image-to-Image Translation Using Conditional Adversarial Networks.
Based on pix2pix by Phillip Isola et al.
The examples from the paper:
Prerequisites
Linux
Python, Numpy, PIL
pytorch 0.4.0
torchvision 0.2.1
Getting Started
Clone this repo:
git clone git@github.com: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
Acknowledgments
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.