Variational Autoencoder (VAE)

This is an implementation of a variational autoencoder using convolutional layers. I will show an example of using it with a sketch dataset.

Dependencies

python train.py -mode train -config configs/sbir_vae.config -name SBIR

python train.py -mode predict -config configs/sbir_vae.config -name SBIR

Name		Name	Last commit message	Last commit date
Latest commit History 7 Commits
configs		configs
models		models
results		results
utils_vae		utils_vae
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
train.py		train.py