• Stars
    star
    119
  • Rank 297,930 (Top 6 %)
  • Language
    Python
  • Created over 7 years ago
  • Updated over 3 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Pytorch implementation of pix2pix for various datasets.

pytorch-pix2pix

Pytorch implementation of pix2pix [1] for various datasets.

dataset

  • cityscapes
    • 2,975 training images, 200 train epochs, 1 batch size, inverse order: True
  • facades
    • 400 training images, 200 train epochs, 1 batch size, inverse order: True
  • maps
    • 1,096 training images, 200 train epochs, 1 batch size, inverse order: True
  • edges2shoes
    • 50k training images, 15 train epochs, 4 batch size, inverse order: False
  • edges2handbags
    • 137k training images, 15 train epochs, 4 batch size, inverse order: False

Resutls

cityscapes

  • cityscapes after 200 epochs
    • First column: input, second column: output, third column: ground truth

city_result

  • Generate animation for fixed inputs

cityscapes_gif

  • Learning Time
    • cityscapes pix2pix - Avg. per epoch: 332.08 sec; Total 200 epochs: 66,846.58 sec

facades

  • facades after 200 epochs
    • First column: input, second column: output, third column: ground truth

facades_result

  • Generate animation for fixed inputs

facades_gif

  • Learning Time
    • facades pix2pix - Avg. per epoch: 44.94 sec; Total 200 epochs: 9,282.64 sec

edges2handbags

  • edges2handbags after 15 epochs
    • First column: input, second column: output, third column: ground truth

edges2handbags_result

  • Generate animation for fixed inputs

edges2handbags_gif

  • Learning Time
    • edges2handbags pix2pix - Avg. per epoch: 10,228.08 sec; Total 15 epochs: 153,443.62 sec

Development Environment

  • Ubuntu 14.04 LTS
  • NVIDIA GTX 1080 ti
  • cuda 8.0
  • Python 2.7.6
  • pytorch 0.1.12
  • matplotlib 1.3.1
  • imageio 2.2.0
  • scipy 0.19.1

Reference

[1] Isola, Phillip, et al. "Image-to-image translation with conditional adversarial networks." arXiv preprint arXiv:1611.07004 (2016).

(Full paper: https://arxiv.org/pdf/1611.07004.pdf)

More Repositories

1

pytorch-generative-model-collections

Collection of generative models in Pytorch version.
Python
2,588
star
2

UGATIT-pytorch

Official PyTorch implementation of U-GAT-IT: Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translation
Python
2,508
star
3

pytorch-MNIST-CelebA-GAN-DCGAN

Pytorch implementation of Generative Adversarial Networks (GAN) and Deep Convolutional Generative Adversarial Networks (DCGAN) for MNIST and CelebA datasets
Python
507
star
4

pytorch-MNIST-CelebA-cGAN-cDCGAN

Pytorch implementation of conditional Generative Adversarial Networks (cGAN) and conditional Deep Convolutional Generative Adversarial Networks (cDCGAN) for MNIST dataset
Python
482
star
5

pytorch-CartoonGAN

Pytorch implementation of CartoonGAN (CVPR 2018)
Python
388
star
6

tensorflow-MNIST-GAN-DCGAN

Tensorflow implementation of Generative Adversarial Networks (GAN) and Deep Convolutional Generative Adversarial Netwokrs for MNIST dataset.
Python
175
star
7

tensorflow-MNIST-cGAN-cDCGAN

Tensorflow implementation of conditional Generative Adversarial Networks (cGAN) and conditional Deep Convolutional Adversarial Networks (cDCGAN) for MANIST dataset.
Python
147
star
8

pytorch-Conditional-image-to-image-translation

Pytorch implementation of Conditional image-to-image translation (CVPR 2018)
Python
48
star
9

pytorch-apex-experiment

Simple experiment of Apex (A PyTorch Extension)
Python
47
star
10

pytorch-CycleGAN

Pytorch implementation of CycleGAN.
Python
40
star
11

FUNIT-pytorch

Pytorch implementation of "Few-Shot Unsupervised Image-to-Image Translation" (ICCV 2019)
Python
36
star
12

tensorflow-pix2pix

Tensorflow implementation of pix2pix for various datasets.
Python
6
star