Exemplar-VAE

Code for reproducing results in Exemplar VAE paper; Accepted to NeurIPS 2020.

Requirements

pip3 install -r requirements.txt

Exemplar VAE Samples

full_generation

Exemplar Based Generation

python3 analysis.py --dir pretrained_model  --generate

celebA_exemplar_generation

exemplar_generation

Density Estimation

python3 density_estimation.py --prior exemplar_prior --dataset {dynamic_mnist|fashion_mnist|omniglot} --model_name {vae|hvae_2level|convhvae_2level} --number_components {25000|11500} --approximate_prior {True|False}

density_estimation

Data Augmentation

python3 analysis.py --dir pretrained_model  --classify

augmentation

data_augmentation

Cyclic Generation

python3 analysis.py --dir pretrained_model  --cyclic_generation

cyclic_generation

GitHub