ABAO77's picture
Upload 161 files
982b011 verified
raw
history blame contribute delete
348 Bytes
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",
]