Skip to content

peria1/VAEconvMNIST

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 

Repository files navigation

Variational Autoencoder for face image generation in PyTorch

Variational Autoencoder for face image generation implemented with PyTorch, Trained over a combination of CelebA + FaceScrub + JAFFE datasets.

Based on Deep Feature Consistent Variational Autoencoder (https://arxiv.org/abs/1610.00291 | https://github.com/houxianxu/DFC-VAE)

TODO: Add DFC-VAE implementation

Pretrained model available at https://drive.google.com/open?id=0B4y-iigc5IzcTlJfYlJyaF9ndlU

Results

Original Faces vs. Reconstructed Faces:

Linear interpolation between two face images:

Vector arithmatic in latent space:

About

Basic Pytorch VAE adapted to use conv2d on MNIST

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages