Skip to content

Latest commit

 

History

History
31 lines (26 loc) · 764 Bytes

README.md

File metadata and controls

31 lines (26 loc) · 764 Bytes
Raw

Retrieval-Augmented Few-shot Image Classification (RAFIC)

Stanford CS330 project

Baseline

Baselines

from rafic import experiments as exp

res = dict()
for dataset in ("birds", "aircraft"):
  res[dataset] = dict(
    zs=exp.zero_shot_text_label(dataset_name=dataset),
    lr=exp.logistic_regression(dataset_name=dataset),
  )

Experiments

bash exp1.sh
bash exp2.sh
bash exp3.sh

Env

  • Install the conda env
  • pip install -r requirements.txt
  • Install CLIP: pip install git+https://github.com/openai/CLIP.git

Data

  • rafic/config.py has a list of all the required data files
  • Download from GDrive and place things in the appropriate folders