config.py

# -----------------
# DATASET ROOTS
# -----------------
cifar_10_root = '../../data/datasets/cifar10'
cifar_100_root = '../../data/datasets/cifar100'
cub_root = '../../data/datasets/CUB'
car_root = '../../data/datasets/stanford_car/cars_{}/'
pets_root = '../../data/datasets/pets/'

aircraft_root = '../../data/datasets/aircraft/fgvc-aircraft-2013b'
herbarium_dataroot = '../../data/datasets/herbarium_19/'
imagenet_root = '../../data/datasets/ImageNet/'#ILSVRC12'

# OSR Split dir
osr_split_dir = '../../data/ssb_splits'

# -----------------
# OTHER PATHS
# -----------------
dino_pretrain_path = '../../pretrained_models/dino/dino_vitbase16_pretrain.pth'
warmup_pretrain_path = '../../pretrained_models/GCDTeacher/model_best.pth'
feature_extract_dir = 'osr_novel_categories/extracted_features_public_impl'     # Extract features to this directory
exp_root = 'osr_novel_categories/'          # All logs and checkpoints will be saved here