Skip to content
No description, website, or topics provided.
Python Shell
Branch: master
Clone or download

Latest commit

Latest commit 8e9894a Apr 1, 2020

Files

Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
dataset fix Apr 1, 2020
distill initial update Mar 24, 2020
eval initial update Mar 24, 2020
models initial update Mar 24, 2020
scripts initial update Mar 24, 2020
.gitignore remove fake evaluation (w/ fixed seed) Mar 26, 2020
README.md Update README.md Mar 26, 2020
eval_fewshot.py fix Apr 1, 2020
train_distillation.py
train_supervised.py remove fake evaluation (w/ fixed seed) Mar 26, 2020
util.py initial update Mar 24, 2020

README.md

RFS

Representations for Few-Shot Learning (RFS). This repo covers the implementation of the following paper:

"Rethinking few-shot image classification: a good embedding is all you need?" Paper, Project Page

If you find this repo useful for your research, please consider citing the paper

@article{tian2020rethink,
  title={Rethinking few-shot image classification: a good embedding is all you need?},
  author={Tian, Yonglong and Wang, Yue and Krishnan, Dilip and Tenenbaum, Joshua B and Isola, Phillip},
  journal={arXiv preprint arXiv:2003.11539},
  year={2020}
}

Installation

This repo was tested with Ubuntu 16.04.5 LTS, Python 3.5, PyTorch 0.4.0, and CUDA 9.0. However, it should be compatible with recent PyTorch versions >=0.4.0

Download Data

The data we used here is preprocessed by the repo of MetaOptNet, but we have renamed the file. Our version of data can be downloaded from here:

[DropBox]

Running

Exemplar commands for running the code can be found in scripts/run.sh.

For unuspervised learning methods CMC and MoCo, please refer to the CMC repo.

Contacts

For any questions, please contact:

Yonglong Tian (yonglong@mit.edu)
Yue Wang (yuewang@csail.mit.edu)

Acknowlegements

Part of the code for distillation is from RepDistiller repo.

You can’t perform that action at this time.