vicellst-att / constants.py
polejowska's picture
Update constants.py
99d6d2f
raw
history blame
662 Bytes
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"]