- Linux/ OSX system with FFmpeg and imagemagick
- Python 3
- Pytorch, Matplotlib, Numpy
Run sudo sh setup.sh
to install and setup the required packages and softwares.
Number of epochs | Dataset | Test Loss | Synthetic Images |
---|---|---|---|
10 | MNIST | 105.23 | |
50 | MNIST | 100.09 | |
20 | Fashion-MNIST | 238.51 |
- Try with CelebA datasets
- Combine with DCGAN (VAE-GAN by Rosca et al.)
- Auto-Encoding Variational Bayes by Kingma et al.
- Stochastic Backpropagation and Approximate Inferencein Deep Generative Models by Rezende et al.
- Junction Tree Variational Autoencoder for Molecular Graph Generation by Jin et al.
- Understand Kullback-Leibler Divergence at here.
- https://github.com/altosaar/variational-autoencoder
- https://jaan.io/what-is-variational-autoencoder-vae-tutorial/