All Projects → znxlwm → Pytorch Mnist Celeba Gan Dcgan

znxlwm / Pytorch Mnist Celeba Gan Dcgan

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

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to Pytorch Mnist Celeba Gan Dcgan

Gan Tutorial
Simple Implementation of many GAN models with PyTorch.
Stars: ✭ 227 (-37.47%)
Mutual labels:  gan, mnist, dcgan, celeba
DCGAN-Pytorch
A Pytorch implementation of "Deep Convolutional Generative Adversarial Networks"
Stars: ✭ 23 (-93.66%)
Mutual labels:  generative-adversarial-network, mnist, dcgan, celeba
Tf Exercise Gan
Tensorflow implementation of different GANs and their comparisions
Stars: ✭ 110 (-69.7%)
Mutual labels:  gan, mnist, dcgan, celeba
Tensorflow Mnist Gan Dcgan
Tensorflow implementation of Generative Adversarial Networks (GAN) and Deep Convolutional Generative Adversarial Netwokrs for MNIST dataset.
Stars: ✭ 163 (-55.1%)
Mutual labels:  gan, generative-adversarial-network, mnist, dcgan
Generative adversarial networks 101
Keras implementations of Generative Adversarial Networks. GANs, DCGAN, CGAN, CCGAN, WGAN and LSGAN models with MNIST and CIFAR-10 datasets.
Stars: ✭ 138 (-61.98%)
Mutual labels:  gan, generative-adversarial-network, mnist, dcgan
Pytorch Generative Model Collections
Collection of generative models in Pytorch version.
Stars: ✭ 2,296 (+532.51%)
Mutual labels:  gan, generative-adversarial-network, mnist
Anogan Tf
Unofficial Tensorflow Implementation of AnoGAN (Anomaly GAN)
Stars: ✭ 218 (-39.94%)
Mutual labels:  gan, generative-adversarial-network, dcgan
Alae
[CVPR2020] Adversarial Latent Autoencoders
Stars: ✭ 3,178 (+775.48%)
Mutual labels:  gan, generative-adversarial-network, celeba
Tensorflow Infogan
🎎 InfoGAN: Interpretable Representation Learning
Stars: ✭ 149 (-58.95%)
Mutual labels:  gan, generative-adversarial-network, mnist
cDCGAN
PyTorch implementation of Conditional Deep Convolutional Generative Adversarial Networks (cDCGAN)
Stars: ✭ 49 (-86.5%)
Mutual labels:  mnist, dcgan, celeba
Dcgan
The Simplest DCGAN Implementation
Stars: ✭ 286 (-21.21%)
Mutual labels:  gan, generative-adversarial-network, dcgan
DCGAN-CelebA-PyTorch-CPP
DCGAN Implementation using PyTorch in both C++ and Python
Stars: ✭ 14 (-96.14%)
Mutual labels:  gan, dcgan, celeba
Image generator
DCGAN image generator 🖼️.
Stars: ✭ 173 (-52.34%)
Mutual labels:  gan, generative-adversarial-network, dcgan
Tensorflow DCGAN
Study Friendly Implementation of DCGAN in Tensorflow
Stars: ✭ 22 (-93.94%)
Mutual labels:  gan, dcgan, celeba
GAN-Project-2018
GAN in Tensorflow to be run via Linux command line
Stars: ✭ 21 (-94.21%)
Mutual labels:  generative-adversarial-network, gan, dcgan
catgan pytorch
Unsupervised and Semi-supervised Learning with Categorical Generative Adversarial Networks
Stars: ✭ 50 (-86.23%)
Mutual labels:  generative-adversarial-network, gan, dcgan
MNIST-invert-color
Invert the color of MNIST images with PyTorch
Stars: ✭ 13 (-96.42%)
Mutual labels:  generative-adversarial-network, gan, mnist
Pytorch Mnist Celeba Cgan Cdcgan
Pytorch implementation of conditional Generative Adversarial Networks (cGAN) and conditional Deep Convolutional Generative Adversarial Networks (cDCGAN) for MNIST dataset
Stars: ✭ 290 (-20.11%)
Mutual labels:  generative-adversarial-network, mnist, celeba
Semantic image inpainting
Semantic Image Inpainting
Stars: ✭ 140 (-61.43%)
Mutual labels:  gan, generative-adversarial-network, dcgan
gan-vae-pretrained-pytorch
Pretrained GANs + VAEs + classifiers for MNIST/CIFAR in pytorch.
Stars: ✭ 134 (-63.09%)
Mutual labels:  generative-adversarial-network, mnist, dcgan

pytorch-MNIST-CelebA-GAN-DCGAN

Pytorch implementation of Generative Adversarial Networks (GAN) [1] and Deep Convolutional Generative Adversarial Networks (DCGAN) [2] for MNIST [3] and CelebA [4] datasets.

  • If you want to train using cropped CelebA dataset, you have to change isCrop = False to isCrop = True.

  • you can download

  • pytorch_CelebA_DCGAN.py requires 64 x 64 size image, so you have to resize CelebA dataset (celebA_data_preprocess.py).

  • pytorch_CelebA_DCGAN.py added learning rate decay code.

Implementation details

  • GAN

GAN

  • DCGAN

Loss

Resutls

MNIST

  • Generate using fixed noise (fixed_z_)
GAN DCGAN
  • MNIST vs Generated images
MNIST GAN after 100 epochs DCGAN after 20 epochs
  • Training loss

    • GAN Loss
  • Learning Time

    • MNIST DCGAN - Avg. per epoch: 197.86 sec; (if you want to reduce learning time, you can change 'generator(128)' and 'discriminator(128)' to 'generator(64)' and 'discriminator(64)' ... then Avg. per epoch: about 67sec in my development environment.)

CelebA

  • Generate using fixed noise (fixed_z_)
DCGAN DCGAN crop
  • CelebA vs Generated images
CelebA DCGAN after 20 epochs DCGAN crop after 30 epochs
  • Learning Time
    • CelebA DCGAN - Avg. per epoch: 732.54 sec; total 20 epochs ptime: 14744.66 sec

Development Environment

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

Reference

[1] Goodfellow, Ian, et al. "Generative adversarial nets." Advances in neural information processing systems. 2014.

(Full paper: http://papers.nips.cc/paper/5423-generative-adversarial-nets.pdf)

[2] Radford, Alec, Luke Metz, and Soumith Chintala. "Unsupervised representation learning with deep convolutional generative adversarial networks." arXiv preprint arXiv:1511.06434 (2015).

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

[3] Y. LeCun, L. Bottou, Y. Bengio, and P. Haffner. "Gradient-based learning applied to document recognition." Proceedings of the IEEE, 86(11):2278-2324, November 1998.

[4] Liu, Ziwei, et al. "Deep learning face attributes in the wild." Proceedings of the IEEE International Conference on Computer Vision. 2015.

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].