77
GitHub - brain-research/acai: Code for "Understanding and Improving Interpo...
source link: https://github.com/brain-research/acai
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.
README.md
Adversarially Constrained Autoencoder Interpolations (ACAI)
Code for the paper "Understanding and Improving Interpolation in Autoencoders via an Adversarial Regularizer" by David Berthelot, Colin Raffel, Aurko Roy, and Ian Goodfellow.
This is not an officially supported Google product.
Setup
Config with virtualenv
sudo apt install virtualenv cd <path_to_code> virtualenv --system-site-packages env2 . env2/bin/activate pip install -r requirements.txt
Config environment variables
Choose a folder where to save the datasets, for example ~/Data
export AE_DATA=~/Data
Installing datasets
python create_datasets.py
Training
CUDA_VISIBLE_DEVICES=0 python acai.py \ --train_dir=TEMP \ --latent=16 --latent_width=2 --depth=16 --dataset=celeba32
All training from the paper can be found in folder runs
.
Models
These are the maintained models:
- aae.py
- acai.py
- baseline.py
- denoising.py
- dropout.py
- vae.py
- vqvae.py
Classifiers / clustering
- classifier_fc.py: fully connected single layer from raw pixels, see
runs/classify.sh
for examples. - Auto-encoder classification is trained at the same as the auto-encoder.
- cluster.py: K-means clustering, see
runs/cluster.sh
for examples.
Utilities
- create_datasets.py: see Installing datsets for more info.
Recommend
About Joyk
Aggregate valuable and interesting links.
Joyk means Joy of geeK