Prepare data:
-
Download MiniImagenet dataset from the Google Drive.
Download Link: https://drive.google.com/open?id=1q9lrXjjFct8RkoBPk8Mbkn1H1mZNFn5U
-
Extract subfolders to './OICEFSL/filelists/miniImagenet/images'.
-
Run script "OICEFFSL/filelists/miniImagenet/DataPreprocessing.sh".
Train:
python ./train.py --dataset miniImagenet --model ResNet18 --method baseline --train_aug
Test:
python ./save_features.py --dataset miniImagenet --model ResNet18 --method baseline --train_aug
python ./test.py --dataset miniImagenet --model ResNet18 --method baseline --train_aug
Our testbed builds upon several existing publicly available code. Specifically, we have modified and integrated the following code into this project:
CloserLookFewShot:https://github.com/wyharveychen/CloserLookFewShot