File size: 600 Bytes
3370ff8
 
 
 
 
 
 
 
 
 
 
9f3ba68
 
 
50b3077
3370ff8
 
50b3077
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 (4)": "polejowska/detr-resnet-101-CD45RB-1000-att-e4",
    "DETR-RESNET-50-NEW": "polejowska/detr-resnet50-cd45rb-subdataset"
}

MODELS_NAMES = ["DETR-RESNET-50", "DETR-RESNET-101", "DETR-RESNET-50 (4)", "DETR-RESNET-50-NEW"]