Official Implementation for "Generalized Zero-Shot Learning via VAE-Conditioned Generative Flow"
├── README.md
├── train_vaepriorflow.py
├── dataset_GBU.py
├── classifier.py
├── scripts/
├── vaeflow/
├── pics
├── data
│ ├── AWA1
│ ├── AWA2
│ ├── CUB
│ └── SUN
-
Download data from Google Cloud
-
Set up conda environment
conda create -n ZSL python=3.6 conda install -c pytorch pytorch torchvision cudatoolkit=10.0 -
Train
cd scripts/ bash train_cub.sh
If you use this code base in your work, please cite
@article{gu2020generalized,
title={Generalized Zero-Shot Learning via VAE-Conditioned Generative Flow},
author={Gu, Yu-Chao and Zhang, Le and Liu, Yun and Lu, Shao-Ping and Cheng, Ming-Ming},
journal={arXiv preprint arXiv:2009.00303},
year={2020}
}