# Semantic Segmentation Classes

def procthor_classes():
    return [
        'apple', 'armchair', 'basketball', 'bed', 'book', 'bottle', 'bowl', 'box', 'cabinet', 
        'chair', 'coffeetable', 'countertop', 'cup', 'desk', 'diningtable', 'dresser', 'fork', 
        'fridge', 'garbagecan', 'houseplant', 'knife', 'laptop', 'microwave', 'ottoman', 'painting', 
        'pillow', 'plate', 'pot', 'shelf', 'sink', 'sofa', 'spoon', 'stool', 'toaster', 'toilet', 'vase',
        'floor', 'wall', 'door', 'window'
    ]

def replica_classes():
    return [
        'basket', 'bed', 'bench', 'bin', 'blanket', 'blinds', 'book', 'bottle', 'box', 'bowl', 'camera', 
        'cabinet', 'chair', 'clock', 'comforter', 'cushion', 'ceiling', 'desk-organizer', 'door', 'floor', 
        'indoor-plant', 'lamp', 'nightstand', 'panel', 'picture', 'pillar', 'pillow', 'plant-stand', 'plate', 
        'pot', 'rack', 'sculpture', 'sofa', 'stool', 'switch', 'table', 'tablet', 'tissue-paper', 'tv-screen', 
        'vase', 'vent', 'wall', 'wall-plug', 'window', 'rug'
    ]

def scannet_pp_classes():
    return [
        'wall', 'ceiling', 'floor', 'door', 'table', 'cabinet', 'light', 'curtain', 'chair', 'shelf',
        'window', 'monitor', 'windowframe', 'sofa', 'bed', 'doorframe', 'wardrobe', 'plant', 'blanket', 'tv',
        'books', 'computer', 'counter', 'bin', 'clothes', 'sink', 'carpet', 'bag', 'fridge', 'picture',
        'pillow', 'cupboard', 'stool', 'suitcase', 'backpack', 'bathtub', 'keyboard', 'toilet', 'printer', 'bench',
        'microwave', 'mattress', 'bottle', 'fan', 'beanbag', 'washing machine', 'telephone', 'stairs', 'dishwasher', 'laptop',
        'pot', 'dresser', 'bicycle', 'staircase', 'mug', 'bike', 'stovetop', 'shirt', 'hair dryer', 'remote',
        'speaker', 'socket',
    ]