├── .gitignore ├── LICENSE ├── README.md ├── data ├── cub │ ├── .gitignore │ └── split │ │ ├── test.csv │ │ ├── train.csv │ │ └── val.csv └── miniimagenet │ ├── .gitignore │ └── split │ ├── aux_test.csv │ ├── aux_val.csv │ ├── test.csv │ ├── train.csv │ └── val.csv ├── eval_fsl.py ├── eval_gfsl.py ├── imgs ├── architecture.png └── gfsl.png ├── model ├── __init__.py ├── dataloader │ ├── mini_imagenet.py │ ├── samplers.py │ └── tiered_imagenet.py ├── evaluator │ ├── __init__.py │ ├── base.py │ ├── fsl_evaluator.py │ ├── gfsl_evaluator.py │ └── helpers.py ├── logger.py ├── models │ ├── __init__.py │ ├── acastle.py │ ├── base.py │ ├── castle.py │ ├── classifier.py │ └── protonet.py ├── networks │ ├── convnet.py │ ├── dropblock.py │ ├── res12.py │ └── resnet.py ├── trainer │ ├── __init__.py │ ├── base.py │ ├── gfsl_trainer.py │ └── helpers.py └── utils.py └── train_gfsl.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/README.md -------------------------------------------------------------------------------- /data/cub/.gitignore: -------------------------------------------------------------------------------- 1 | images 2 | -------------------------------------------------------------------------------- /data/cub/split/test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/data/cub/split/test.csv -------------------------------------------------------------------------------- /data/cub/split/train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/data/cub/split/train.csv -------------------------------------------------------------------------------- /data/cub/split/val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/data/cub/split/val.csv -------------------------------------------------------------------------------- /data/miniimagenet/.gitignore: -------------------------------------------------------------------------------- 1 | images 2 | -------------------------------------------------------------------------------- /data/miniimagenet/split/aux_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/data/miniimagenet/split/aux_test.csv -------------------------------------------------------------------------------- /data/miniimagenet/split/aux_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/data/miniimagenet/split/aux_val.csv -------------------------------------------------------------------------------- /data/miniimagenet/split/test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/data/miniimagenet/split/test.csv -------------------------------------------------------------------------------- /data/miniimagenet/split/train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/data/miniimagenet/split/train.csv -------------------------------------------------------------------------------- /data/miniimagenet/split/val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/data/miniimagenet/split/val.csv -------------------------------------------------------------------------------- /eval_fsl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/eval_fsl.py -------------------------------------------------------------------------------- /eval_gfsl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/eval_gfsl.py -------------------------------------------------------------------------------- /imgs/architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/imgs/architecture.png -------------------------------------------------------------------------------- /imgs/gfsl.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/imgs/gfsl.png -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/dataloader/mini_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/dataloader/mini_imagenet.py -------------------------------------------------------------------------------- /model/dataloader/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/dataloader/samplers.py -------------------------------------------------------------------------------- /model/dataloader/tiered_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/dataloader/tiered_imagenet.py -------------------------------------------------------------------------------- /model/evaluator/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/evaluator/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/evaluator/base.py -------------------------------------------------------------------------------- /model/evaluator/fsl_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/evaluator/fsl_evaluator.py -------------------------------------------------------------------------------- /model/evaluator/gfsl_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/evaluator/gfsl_evaluator.py -------------------------------------------------------------------------------- /model/evaluator/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/evaluator/helpers.py -------------------------------------------------------------------------------- /model/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/logger.py -------------------------------------------------------------------------------- /model/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/models/__init__.py -------------------------------------------------------------------------------- /model/models/acastle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/models/acastle.py -------------------------------------------------------------------------------- /model/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/models/base.py -------------------------------------------------------------------------------- /model/models/castle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/models/castle.py -------------------------------------------------------------------------------- /model/models/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/models/classifier.py -------------------------------------------------------------------------------- /model/models/protonet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/models/protonet.py -------------------------------------------------------------------------------- /model/networks/convnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/networks/convnet.py -------------------------------------------------------------------------------- /model/networks/dropblock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/networks/dropblock.py -------------------------------------------------------------------------------- /model/networks/res12.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/networks/res12.py -------------------------------------------------------------------------------- /model/networks/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/networks/resnet.py -------------------------------------------------------------------------------- /model/trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/trainer/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/trainer/base.py -------------------------------------------------------------------------------- /model/trainer/gfsl_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/trainer/gfsl_trainer.py -------------------------------------------------------------------------------- /model/trainer/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/trainer/helpers.py -------------------------------------------------------------------------------- /model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/model/utils.py -------------------------------------------------------------------------------- /train_gfsl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sha-Lab/aCASTLE/HEAD/train_gfsl.py --------------------------------------------------------------------------------