Minimalist Pix2Pix implementation in pytorch. Paper: https://arxiv.org/abs/1611.07004
The aim of this implementation is easy experimenting.
We followed the paper where possible and filled the blanks with the official implementation. The code may not be compatible with the paper 100%. Fixes and suggestions are welcome.
- Download cityscapes dataset and extract it.
- Run the script with
python3 pix2pix.py cityscapes
That's it.
No augmentations implemented. Default direction is B to A because that's how cityscapes dataset is. You can change it in load_image function.
Here's an example output after 50 epochs. The output images are always from validation set.
