An open API service indexing awesome lists of open source software.

https://github.com/lyeoni/pytorch-mnist-gan


https://github.com/lyeoni/pytorch-mnist-gan

deep-learning gan generative-adversarial-network generative-model mnist pytorch

Last synced: about 1 year ago
JSON representation

Awesome Lists containing this project

README

          

# pytorch-mnist-GAN
Pytorch implementation of GAN(Generative Adversarial Networks) on the MNIST data set

## Dependencies
- PyTorch
- torchvision
- numpy

## Results
| Mode collapsing | After 100 epochs | After 200 epochs |
|:--------:|:--------:|:--------:|
|![alt text](https://github.com/lyeoni/pytorch-mnist-GAN/blob/master/samples/mode_collapsing.png) | ![alt text](https://github.com/lyeoni/pytorch-mnist-GAN/blob/master/samples/sample1_.png) |![alt text](https://github.com/lyeoni/pytorch-mnist-GAN/blob/master/samples/sample2_.png)|

## Reference
1. Goodfellow, Ian, et al. "Generative Adversarial Networks":
https://arxiv.org/pdf/1406.2661.pdf
2. znxlwm/pytorch-MNIST-CelebA-GAN-DCGAN (github):
https://github.com/znxlwm/pytorch-MNIST-CelebA-GAN-DCGAN
3. prcastro/pytorch-gan (github):
https://github.com/prcastro/pytorch-gan