Hereby we present plain VAE and modified VAE model, both of which are trained on celebA dataset to synthesize facial images.
input image:
To run the code, you are required to install Tensorflow and Tensorlayer on your machine. how to install Tensorlayer
This is the code for the paper Deep Feature Consistent Variational Autoencoder
In loss function we used a vgg loss.Check this how to load and use a pretrained VGG-16? if you have trouble reading vgg_loss.py.
Firstly, download the celebA dataset and VGG-16 weights. After installing all the third-party packages required, we can train the models by:
python train_vae.py # for plain VAE
python train_dfc_vae.py # for DFC-VAE