VAE

This repository is for sharing the scripts of VAE.

Description

You can train MNIST with any number of latent variable dimensions, and visualize the latent space from multiple perspectives.

Requirement

• Python: 3.7
• fire: 0.4

Usage

You can use the following command to train and visualize VAE.

python main.py main --z_dim [The dimensions of latent variables]

You can start tensorboard with the following command. Note that The default URL is http://localhost:6006/.

tensorboard --logdir ./logs

Network

The configuration of VAE is shown as below.

Output

You can get the following images.

Reconstruction

Mappings of the latent space

Generation from lattice points

Walkthrough