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

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


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

autoencoder cvae deep-learning generative-model mnist pytorch

Last synced: 11 months ago
JSON representation

Awesome Lists containing this project

README

          

# pytorch-mnist-CVAE
Conditional Variational AutoEncoder on the MNIST data set using the PyTroch

## Dependencies
- PyTorch
- torchvision
- numpy

## Results
Learned MNIST manifold with a condition of label (from 0 to 9)

![alt text](https://github.com/lyeoni/pytorch-mnist-CVAE/blob/master/samples/sample_.png)

## Reference
1. Learning Structured Output Representation using Deep Conditional Generative Models (paper):
https://papers.nips.cc/paper/5775-learning-structured-output-representation-using-deep-conditional-generative-modelsd
2. jojonki/AutoEncoders (github):
https://github.com/jojonki/AutoEncoders/blob/master/cvae.ipynb
3. hujinsen/pytorch_VAE_CVAE (github):
https://github.com/hujinsen/pytorch_VAE_CVAE/blob/master/CVAE.ipynb
4. Prasanna1991/pytorch-vae (github):
https://github.com/Prasanna1991/pytorch-vae/blob/master/cvae.py