File size: 662 Bytes
3370ff8
 
 
 
 
 
 
 
 
 
 
9f3ba68
 
3370ff8
9f3ba68
99d6d2f
3370ff8
 
99d6d2f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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",
    "YOLOS-TINY": "polejowska/yolos-tiny-CD45RB-1000-att",
    "DETR-RESNET-50 (4)": "polejowska/detr-resnet-101-CD45RB-1000-att-e4",
    "DETR-RESNET-50-NEW": "polejowska/detr-resnet50-cd45rb"
}

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