Linyiqi 3508afca93
Reformat config files (#46)
* fix init

* fix test api

fix test api bug

* add metarcnn fsdetview config

* update pr

* update meta file

* fix CI
2021-11-01 23:27:01 +08:00

3.9 KiB

#

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_plus/cub/baseline-plus_conv4_1xb64_cub_5way-1shot.py

# meta testing
python ./tools/classification/test.py \
  configs/classification/baseline_plus/cub/baseline-plus_conv4_1xb64_cub_5way-1shot.py \
  work_dir/baseline-plus_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 1000 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 1000 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 1000 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