pixano.datasets.utils.labels

coco_ids_80to91(category_id)

Return COCO category ID (80 to 91 classes).

Parameters:

Name	Type	Description	Default
category_id	int	Category ID (80 classes).	required

Returns:

Type	Description
int	Category ID (91 classes).

Source code in pixano/datasets/utils/labels.py

def coco_ids_80to91(category_id: int) -> int:
    """Return COCO category ID (80 to 91 classes).

    Args:
        category_id: Category ID (80 classes).

    Returns:
        Category ID (91 classes).
    """
    coco_dict = {
        1: 1,
        2: 2,
        3: 3,
        4: 4,
        5: 5,
        6: 6,
        7: 7,
        8: 8,
        9: 9,
        10: 10,
        11: 11,
        12: 13,
        13: 14,
        14: 15,
        15: 16,
        16: 17,
        17: 18,
        18: 19,
        19: 20,
        20: 21,
        21: 22,
        22: 23,
        23: 24,
        24: 25,
        25: 27,
        26: 28,
        27: 31,
        28: 32,
        29: 33,
        30: 34,
        31: 35,
        32: 36,
        33: 37,
        34: 38,
        35: 39,
        36: 40,
        37: 41,
        38: 42,
        39: 43,
        40: 44,
        41: 46,
        42: 47,
        43: 48,
        44: 49,
        45: 50,
        46: 51,
        47: 52,
        48: 53,
        49: 54,
        50: 55,
        51: 56,
        52: 57,
        53: 58,
        54: 59,
        55: 60,
        56: 61,
        57: 62,
        58: 63,
        59: 64,
        60: 65,
        61: 67,
        62: 70,
        63: 72,
        64: 73,
        65: 74,
        66: 75,
        67: 76,
        68: 77,
        69: 78,
        70: 79,
        71: 80,
        72: 81,
        73: 82,
        74: 84,
        75: 85,
        76: 86,
        77: 87,
        78: 88,
        79: 89,
        80: 90,
    }

    return coco_dict[int(category_id)]

coco_names_80(category_id)

Return COCO category name (80 classes).

Parameters:

Name	Type	Description	Default
category_id	int	Category ID.	required

Returns:

Type	Description
str	Category name.

Source code in pixano/datasets/utils/labels.py

def coco_names_80(category_id: int) -> str:
    """Return COCO category name (80 classes).

    Args:
        category_id: Category ID.

    Returns:
        Category name.
    """
    coco_dict = {
        1: "person",
        2: "bicycle",
        3: "car",
        4: "motorcycle",
        5: "airplane",
        6: "bus",
        7: "train",
        8: "truck",
        9: "boat",
        10: "traffic light",
        11: "fire hydrant",
        12: "stop sign",
        13: "parking meter",
        14: "bench",
        15: "bird",
        16: "cat",
        17: "dog",
        18: "horse",
        19: "sheep",
        20: "cow",
        21: "elephant",
        22: "bear",
        23: "zebra",
        24: "giraffe",
        25: "backpack",
        26: "umbrella",
        27: "handbag",
        28: "tie",
        29: "suitcase",
        30: "frisbee",
        31: "skis",
        32: "snowboard",
        33: "sports ball",
        34: "kite",
        35: "baseball bat",
        36: "baseball glove",
        37: "skateboard",
        38: "surfboard",
        39: "tennis racket",
        40: "bottle",
        41: "wine glass",
        42: "cup",
        43: "fork",
        44: "knife",
        45: "spoon",
        46: "bowl",
        47: "banana",
        48: "apple",
        49: "sandwich",
        50: "orange",
        51: "broccoli",
        52: "carrot",
        53: "hot dog",
        54: "pizza",
        55: "donut",
        56: "cake",
        57: "chair",
        58: "couch",
        59: "potted plant",
        60: "bed",
        61: "dining table",
        62: "toilet",
        63: "tv",
        64: "laptop",
        65: "mouse",
        66: "remote",
        67: "keyboard",
        68: "cell phone",
        69: "microwave",
        70: "oven",
        71: "toaster",
        72: "sink",
        73: "refrigerator",
        74: "book",
        75: "clock",
        76: "vase",
        77: "scissors",
        78: "teddy bear",
        79: "hair drier",
        80: "toothbrush",
    }

    return coco_dict[int(category_id)]

coco_names_91(cat_id)

Return COCO category name (91 classes).

Parameters:

Name	Type	Description	Default
cat_id	int	Category ID.	required

Returns:

Type	Description
str	Category name.

Source code in pixano/datasets/utils/labels.py

def coco_names_91(cat_id: int) -> str:
    """Return COCO category name (91 classes).

    Args:
        cat_id: Category ID.

    Returns:
        Category name.
    """
    coco_dict = {
        1: "person",
        2: "bicycle",
        3: "car",
        4: "motorcycle",
        5: "airplane",
        6: "bus",
        7: "train",
        8: "truck",
        9: "boat",
        10: "traffic light",
        11: "fire hydrant",
        12: "street sign",
        13: "stop sign",
        14: "parking meter",
        15: "bench",
        16: "bird",
        17: "cat",
        18: "dog",
        19: "horse",
        20: "sheep",
        21: "cow",
        22: "elephant",
        23: "bear",
        24: "zebra",
        25: "giraffe",
        26: "hat",
        27: "backpack",
        28: "umbrella",
        29: "shoe",
        30: "eye glasses",
        31: "handbag",
        32: "tie",
        33: "suitcase",
        34: "frisbee",
        35: "skis",
        36: "snowboard",
        37: "sports ball",
        38: "kite",
        39: "baseball bat",
        40: "baseball glove",
        41: "skateboard",
        42: "surfboard",
        43: "tennis racket",
        44: "bottle",
        45: "plate",
        46: "wine glass",
        47: "cup",
        48: "fork",
        49: "knife",
        50: "spoon",
        51: "bowl",
        52: "banana",
        53: "apple",
        54: "sandwich",
        55: "orange",
        56: "broccoli",
        57: "carrot",
        58: "hot dog",
        59: "pizza",
        60: "donut",
        61: "cake",
        62: "chair",
        63: "couch",
        64: "potted plant",
        65: "bed",
        66: "mirror",
        67: "dining table",
        68: "window",
        69: "desk",
        70: "toilet",
        71: "door",
        72: "tv",
        73: "laptop",
        74: "mouse",
        75: "remote",
        76: "keyboard",
        77: "cell phone",
        78: "microwave",
        79: "oven",
        80: "toaster",
        81: "sink",
        82: "refrigerator",
        83: "blender",
        84: "book",
        85: "clock",
        86: "vase",
        87: "scissors",
        88: "teddy bear",
        89: "hair drier",
        90: "toothbrush",
        91: "hair brush",
    }

    return coco_dict[int(cat_id)]

dota_ids(name)

Return DOTAv2 category ID (18 classes).

Parameters:

Name	Type	Description	Default
name	str	Category name.	required

Returns:

Type	Description
int	Category ID.

Source code in pixano/datasets/utils/labels.py

def dota_ids(name: str) -> int:
    """Return DOTAv2 category ID (18 classes).

    Args:
        name: Category name.

    Returns:
        Category ID.
    """
    dota_dict = {
        "plane": 1,
        "ship": 2,
        "storage tank": 3,
        "baseball diamond": 4,
        "tennis court": 5,
        "basketball court": 6,
        "ground track field": 7,
        "harbor": 8,
        "bridge": 9,
        "large vehicle": 10,
        "small vehicle": 11,
        "helicopter": 12,
        "roundabout": 13,
        "soccer ball field": 14,
        "swimming pool": 15,
        "container crane": 16,
        "airport": 17,
        "helipad": 18,
    }

    return dota_dict[str(name).replace("-", " ")]

voc_names(category_id)

Return VOC category name (20 classes).

Parameters:

Name	Type	Description	Default
category_id	int	Category ID.	required

Returns:

Type	Description
str	Category name.

Source code in pixano/datasets/utils/labels.py

def voc_names(category_id: int) -> str:
    """Return VOC category name (20 classes).

    Args:
        category_id: Category ID.

    Returns:
        Category name.
    """
    voc_dict = {
        1: "aeroplane",
        2: "bicycle",
        3: "bird",
        4: "boat",
        5: "bottle",
        6: "bus",
        7: "car",
        8: "cat",
        9: "chair",
        10: "cow",
        11: "dining table",
        12: "dog",
        13: "horse",
        14: "motorbike",
        15: "person",
        16: "potted plant",
        17: "sheep",
        18: "sofa",
        19: "train",
        20: "tv / monitor",
    }

    return voc_dict[int(category_id)]