- Python 3.6+
- PyTorch 0.4.0+
- qpth 0.0.11+
- tqdm
- kornia
- Clone this repository:
git clone https://github.com/erictang000/meta_instance_aug.git
- Download and decompress dataset files: miniImageNet (courtesy of Spyros Gidaris), CIFAR-FS
- All experiment commands for training instance specific augmentations can be found in bash.sh.
- Test time augmentation was added in test_aug.ipynb, and results can be reproduced in that notebook.
This code is based off the of the MetaAug and InstaAug repos.