Skip to content

An exercise project that uses sequence VAE to encode and reconstruct mnist images (implemented in Keras)

Notifications You must be signed in to change notification settings

sicong-huang/rnn-vae-mnist-model

Repository files navigation

An RNN based VAE for MNIST data

This repository is an exercise project using a sequence variational auto-encoder (VAE) to encode and generate MNIST hand-written digits.

A simple script to interpolate between latent representations of 2 mnist digits is also included.

Model Architecture

image1

Dependencies

Other than python built-in modules, the following are additional required dependicies

  • numpy
  • keras==2.2.2
  • tensorflow==1.10.0
  • matplotlib==2.1.2

Usage

The model is implemented in file model.py, and the code to train is written in train.py.

python train.py  # pass -h option for more options

After running train.py, trained model is saved in saved_models/

To interpolate between digits, run

python interpolate.py  # pass -h option for more options

About

An exercise project that uses sequence VAE to encode and reconstruct mnist images (implemented in Keras)

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages