src/config/settings.py

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",
]