constants.py

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",
}

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