Variational_Transfer_Fewshot

Implementation for Variational Transfer Learning for Fine-grained Few-shot Visual Recognition

Environment

Python3
Pytorch 1.4.0

Datasets

CUB

Change directoy to filelists/CUB
Download the dataset from http://www.vision.caltech.edu/visipedia/CUB-200-2011.html
Put it as ./CUB_200_2011
Run python ./write_CUB_filelist.py

NAB

Change directoy to filelists/NAB
Download the dataset from http://dl.allaboutbirds.org/nabirds
Put it as ./nabirds
Run python ./write_NAB_filelist.py

Stanford Dogs

Change directoy to filelists/DOG
Download the dataset from http://vision.stanford.edu/aditya86/ImageNetDogs/
Put it as ./stanforddogs
Run python ./write_DOG_filelist.py

Running Experiments

Run training phase:

python train_vae.py

Run testing phase:

python finetune_sample.py