File size: 348 Bytes
982b011
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
WORK_DIR = "./"
DATA_DIR = f"{WORK_DIR}/data"
IMAGES_DIR = f"{DATA_DIR}/images"
RESULTS_DIR = f"{WORK_DIR}/results"

# supported feature extractor models
FEATURE_EXTRACTOR_MODELS = [
    "resnet18",
    "resnet34",
    "resnet50",
    "resnet101",
    "resnet152",
    "vit_b_16",
    "vit_b_32",
    "vit_l_16",
    "vit_l_32",
    "vit_h_14",
]