File size: 493 Bytes
3370ff8
 
 
 
 
 
 
 
 
 
 
9f3ba68
 
090197f
3370ff8
 
9b91786
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
COLORS = [
    [0.000, 0.447, 0.741],
    [0.850, 0.325, 0.098],
    [0.929, 0.694, 0.125],
    [0.494, 0.184, 0.556],
    [0.466, 0.674, 0.188],
    [0.301, 0.745, 0.933],
    [0.351, 0.760, 0.903],
]

MODELS_REPO = {
    "DETR-RESNET-50": "polejowska/detr-resnet-50-CD45RB-1000-att",
    "DETR-RESNET-101": "polejowska/detr-resnet-101-CD45RB-1000-att",
    "DETR-RESNET-50-TEST": "polejowska/detr-resnet50-leuk"
}

MODELS_NAMES = ["DETR-RESNET-50", "DETR-RESNET-101", "DETR-RESNET-50-TEST"]