pixano.datasets.utils.labels

category_id(category_name, category_dict='coco91')

Return category ID based on category name.

Parameters:

Name	Type	Description	Default
category_name	str	Category name	required
category_dict	str	Which category dictionary to use ("coco91", "coco80", "voc")	'coco91'

Returns:

Type	Description
int	Category ID

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

def category_id(category_name: str, category_dict: str = "coco91") -> int:
    """Return category ID based on category name.

    Args:
        category_name: Category name
        category_dict: Which category dictionary to use ("coco91", "coco80", "voc")

    Returns:
        Category ID
    """
    return CATEGORY_IDS[category_dict][str(category_name).strip().lower()]

category_name(category_id, category_dict='coco91')

Return category name based on category ID.

Parameters:

Name	Type	Description	Default
category_id	int	Category ID	required
category_dict	str	Which category dictionary to use ("coco91", "coco80", "voc")	'coco91'

Returns:

Type	Description
str	Category name

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

def category_name(category_id: int, category_dict: str = "coco91") -> str:
    """Return category name based on category ID.

    Args:
        category_id: Category ID
        category_dict: Which category dictionary to use ("coco91", "coco80", "voc")

    Returns:
        Category name
    """
    return CATEGORY_NAMES[category_dict][int(category_id)]

coco_ids_80to91(category_id)

Convert COCO category ID from 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:
    """Convert COCO category ID from 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)]