mirror of
https://github.com/open-mmlab/mmfewshot.git
synced 2025-06-03 14:49:43 +08:00
* fix init * fix test api fix test api bug * add metarcnn fsdetview config * update pr * update meta file * fix CI
Baseline (ICLR'2019)
@inproceedings{chen2019closerfewshot,
title={A Closer Look at Few-shot Classification},
author={Chen, Wei-Yu and Liu, Yen-Cheng and Kira, Zsolt and Wang, Yu-Chiang and Huang, Jia-Bin},
booktitle={International Conference on Learning Representations},
year={2019}
}
How to Reproduce Baseline
It consists of two steps:
-
Step1: Base training
- use all the images of base classes to train a base model.
- use validation set to select the best model.
-
Step2: Meta Testing:
- use best model from step1.
An example of CUB dataset with Conv4
# base training
python ./tools/classification/train.py \
configs/classification/baseline/cub/baseline_conv4_1xb64_cub_5way-1shot.py
# meta testing
python ./tools/classification/test.py \
configs/classification/baseline/cub/baseline_conv4_1xb64_cub_5way-1shot.py \
work_dir/baseline_conv4_1xb64_cub_5way-1shot/best_accuracy_mean.pth
Note:
- All the result are trained with single gpu.
- The base training of 1 shot and 5 shot use same training setting, but different validation setting.
Results on CUB dataset of 2000 episodes
Arch | Input Size | Batch Size | way | shot | mean Acc | std | ckpt | log |
---|---|---|---|---|---|---|---|---|
conv4 | 84x84 | 64 | 5 | 1 | - | - | ckpt | log |
conv4 | 84x84 | 64 | 5 | 5 | - | - | ckpt | log |
resnet12 | 84x84 | 64 | 5 | 1 | - | - | ckpt | log |
resnet12 | 84x84 | 64 | 5 | 5 | - | - | ckpt | log |
Results on Mini-ImageNet dataset of 2000 episodes
Arch | Input Size | Batch Size | way | shot | mean Acc | std | ckpt | log |
---|---|---|---|---|---|---|---|---|
conv4 | 84x84 | 64 | 5 | 1 | - | - | ckpt | log |
conv4 | 84x84 | 64 | 5 | 5 | - | - | ckpt | log |
resnet12 | 84x84 | 64 | 5 | 1 | - | - | ckpt | log |
resnet12 | 84x84 | 64 | 5 | 5 | - | - | ckpt | log |
Results on Tiered-ImageNet of 2000 episodes
Arch | Input Size | Batch Size | way | shot | mean Acc | std | ckpt | log |
---|---|---|---|---|---|---|---|---|
conv4 | 84x84 | 64 | 5 | 1 | - | - | ckpt | log |
conv4 | 84x84 | 64 | 5 | 5 | - | - | ckpt | log |
resnet12 | 84x84 | 64 | 5 | 1 | - | - | ckpt | log |
resnet12 | 84x84 | 64 | 5 | 5 | - | - | ckpt | log |