gfringeli / vae

VAE implementation for 2D shape dataset

Variational Auto-Encoder and its Variants

This project contains an implementation of the VAE models, and their variants used in Disentangling by Factorising. Hyunjik Kim, Andriy Mnih. 2019.

Usage

• Download dSprites dataset, and store the .npz file under data/2d_sprites.npz.
• Use python vae.py --help for information on how to use the accompanying script.