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:

examples

Prerequisites

  • Linux
  • Python, Numpy, PIL
  • pytorch 0.4.0
  • torchvision 0.2.1

Getting Started

  • 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

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 by Jun-Yan Zhu.

GitHub

GitHub - mrzhu-cool/pix2pix-pytorch: PyTorch implementation of “Image-to-Image Translation Using Conditional Adversarial Networks”.
PyTorch implementation of "Image-to-Image Translation Using Conditional Adversarial Networks". - GitHub - mrzhu-cool/pix2pix-pytorch: PyTorch implementation of "Image-to-Image Transl...