Spaces:
Sleeping
Sleeping
File size: 960 Bytes
df9c255 33ce126 df9c255 cd71529 df9c255 cd71529 df9c255 cd71529 df9c255 cd71529 df9c255 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
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"
]
|