implement of paper:Multi-pretext Attention Network for Few-shot Learning with Self-supervision in Pytorch
conda version:4.6.14
python:3.7.4
pytorch:1.3.0
cuda:10.2
pip install -e .
To download the MiniImagenet dataset go to https://github.com/gidariss/FewShotWithoutForgetting and follow the instructions there. Then, set in man/datasets/mini_imagenet.py the path to where the dataset resides in your machine.
sh run_miniImageNet/WRN-train.sh
sh run_miniImageNet/WRN-test.sh
@article{li2021multi,
title={Multi-Pretext Attention Network for Few-shot Learning with Self-supervision},
author={Li, Hainan and Tao, Renshuai and Li, Jun and Qin, Haotong and Ding, Yifu and Wang, Shuo and Liu, Xianglong},
journal={arXiv preprint arXiv:2103.05985},
year={2021}
}