Skip to content

Code for the models in "PixelVAE: A Latent Variable Model for Natural Images" (https://arxiv.org/abs/1611.05013

Notifications You must be signed in to change notification settings

igul222/PixelVAE

Repository files navigation

PixelVAE

Code for the models in PixelVAE: A Latent Variable Model for Natural Images

Binarized MNIST

To train:

python models/mnist_pixelvae_train.py -L 12 -fs 5 -algo cond_z_bias -dpx 16 -ldim 16

To evaluate, take the weights of the model with best validation score from the above training procedure and then run

python models/mnist_pixelvae_evaluate.py -L 12 -fs 5 -algo cond_z_bias -dpx 16 -ldim 16 -w path/to/weights.pkl

Real-valued MNIST, LSUN Bedrooms, 64x64 ImageNet

To train, evaluate, and generate samples:

python pixelvae.py

By default, this runs on real-valued MNIST. You can specify different datasets or model settings within pixelvae.py.

About

Code for the models in "PixelVAE: A Latent Variable Model for Natural Images" (https://arxiv.org/abs/1611.05013

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages