"""Leaderboard data for different tasks."""

CLASSIFICATION_DATA = {
    "Model": [
        "ResNet-50",
        "ViT-Base",
        "Swin-T",
        "InceptionV3",
        "SqueezeNet",
        "ResNet-50",
        "ViT-Base",
        "Swin-T",
        "InceptionV3",
        "SqueezeNet",
        "ResNet-50",
        "ViT-Base",
        "Swin-T",
        "InceptionV3",
        "SqueezeNet",
    ],
    "Dataset": [
        "DoMars16",
        "DoMars16",
        "DoMars16",
        "DoMars16",
        "DoMars16",
        "Atmospheric Dust",
        "Atmospheric Dust",
        "Atmospheric Dust",
        "Atmospheric Dust",
        "Atmospheric Dust",
        "Martian Frost",
        "Martian Frost",
        "Martian Frost",
        "Martian Frost",
        "Martian Frost",
    ],
    "Accuracy": [
        92.5, 94.2, 95.8, 93.1, 89.7,
        88.3, 90.1, 91.5, 89.8, 87.2,
        85.6, 87.9, 88.4, 86.7, 84.3,
    ],
    "F1-Score": [
        91.8, 93.5, 94.9, 92.4, 88.6,
        87.5, 89.2, 90.7, 88.9, 86.3,
        84.8, 86.9, 87.5, 85.8, 83.4,
    ],
}

DETECTION_DATA = {
    "Model": [
        "Faster R-CNN",
        "YOLOv5",
        "DETR",
        "RetinaNet",
        "SSD",
        "Faster R-CNN",
        "YOLOv5",
        "DETR",
        "RetinaNet",
        "SSD",
        "Faster R-CNN",
        "YOLOv5",
        "DETR",
        "RetinaNet",
        "SSD",
    ],
    "Dataset": [
        "Mars Crater",
        "Mars Crater",
        "Mars Crater",
        "Mars Crater",
        "Mars Crater",
        "Rover Component",
        "Rover Component",
        "Rover Component",
        "Rover Component",
        "Rover Component",
        "Geological Feature",
        "Geological Feature",
        "Geological Feature",
        "Geological Feature",
        "Geological Feature",
    ],
    "mAP": [
        78.5, 80.2, 82.1, 79.3, 77.8,
        75.6, 77.3, 78.9, 76.7, 75.1,
        73.4, 75.1, 76.7, 74.5, 73.0,
    ],
    "IoU": [
        0.72, 0.74, 0.76, 0.73, 0.71,
        0.69, 0.71, 0.73, 0.70, 0.68,
        0.67, 0.69, 0.71, 0.68, 0.67,
    ],
}

SEGMENTATION_DATA = {
    "Model": [
        "U-Net",
        "DeepLabV3+",
        "Mask R-CNN",
        "SegFormer",
        "HRNet",
        "U-Net",
        "DeepLabV3+",
        "Mask R-CNN",
        "SegFormer",
        "HRNet",
        "U-Net",
        "DeepLabV3+",
        "Mask R-CNN",
        "SegFormer",
        "HRNet",
    ],
    "Dataset": [
        "Mars Terrain",
        "Mars Terrain",
        "Mars Terrain",
        "Mars Terrain",
        "Mars Terrain",
        "Dust Storm",
        "Dust Storm",
        "Dust Storm",
        "Dust Storm",
        "Dust Storm",
        "Geological Feature",
        "Geological Feature",
        "Geological Feature",
        "Geological Feature",
        "Geological Feature",
    ],
    "Dice Score": [
        0.85, 0.87, 0.88, 0.86, 0.84,
        0.82, 0.84, 0.85, 0.83, 0.82,
        0.81, 0.83, 0.84, 0.82, 0.81,
    ],
    "IoU": [
        0.74, 0.76, 0.78, 0.75, 0.73,
        0.70, 0.72, 0.74, 0.71, 0.70,
        0.68, 0.70, 0.72, 0.69, 0.68,
    ],
}

TASK_DATA = {
    "Classification": (CLASSIFICATION_DATA, ["Accuracy", "F1-Score"]),
    "Object Detection": (DETECTION_DATA, ["mAP", "IoU"]),
    "Segmentation": (SEGMENTATION_DATA, ["Dice Score", "IoU"]),
}