src/utils.py

MODEL_BASE_PATH = "models"

MODELS = {
    "densenet121": {
        "experiment_name": "05-change-classifier-head",
        "folder": "densenet121_imagenet_1k_adamw_32_bce_aug_class/model.pth.tar",
        "ckpt_path": f"{MODEL_BASE_PATH}/densenet121-in1k.pth.tar"
    },
    "swin_in22k": {
        "experiment_name": "06-transformers-pre-trained",
        "ckpt_path": f"{MODEL_BASE_PATH}/swin-in22k.pth.tar"

    },
    "vit_in1k": {
        "experiment_name": "06-transformers-pre-trained",
        "ckpt_path": f"{MODEL_BASE_PATH}/vit-in1k.pth.tar"
    },
    "swin_simim": {
        "experiment_name": "07-transformer-ssl",
        "ckpt_path": f"{MODEL_BASE_PATH}/swin-simmim.pth.tar"
    },
}

LABELS = [
    "Atelectasis",
    "Cardiomegaly",
    "Consolidation",
    "Edema",
    "Effusion",
    "Emphysema",
    "Fibrosis",
    "Hernia",
    "Infiltration",
    "Mass",
    "Nodule",
    "Pleural Thickening",
    "Pneumonia",
    "Pneumothorax"
]