Function get_class_names

semantic_sam/utils/misc.py:177–242

(name, background=True)

def get_class_names(name, background=True):
    if name is None:
        return None
    if 'refcoco' in name:
        class_names = ["noun"]
    elif 'pascal' in name:
        class_names = PASCAL_CLASSES_PART + ["background"]
    elif 'sam' in name:
        class_names = ['foreground'] + ["background"]
    elif 'coco' in name and 'pan' not in name:
        class_names = COCO_INSTANCE_CLASSES + ["background"]
    elif 'coco' in name:
        class_names = COCO_PANOPTIC_CLASSES + ["background"]
    elif 'ade20k_full' in name:
        class_names = ADE20K_847 + ["background"]
    elif 'ade' in name:
        class_names = ADE_PANOPTIC_CLASSES + ["background"]
    elif 'voc' in name:
        class_names = PASCAL_CLASSES + ["background"]
    elif 'vlp' in name:
        class_names = ["noun"]
    elif 'tsv' in name:
        class_names = ["noun"]
    elif 'phrasecut' in name:
        class_names = ["noun"]
    elif 'openimage' in name:
        class_names = ["noun"]
    elif 'imagenet' in name:
        class_names = IMAGENET_CLASSES
    elif 'context_459' in name:
        class_names = PASCAL_CONTEXT_459 + ["background"]
    elif 'context_59' in name:
        class_names = PASCAL_CONTEXT_59 + ["background"]
    elif 'context_33' in name:
        class_names = PASCAL_CONTEXT_33
    elif 'sunrgbd_37' in name:
        class_names = SUN_RGBD_37
    elif 'scannet_41' in name:
        class_names = SCAN_40
    elif 'scannet_38' in name:
        class_names = SCAN_37
    elif 'scannet_21' in name:
        class_names = SCAN_20
    elif 'object365' in name:
        class_names = OBJECT365
    elif 'lvis' in name:
        class_names = LVIS_CATEGORIES
    elif 'seginw' in name:
        class_names = SEGINW_CATEGORIES[name.replace('_train', '').replace('_val', '')] + ["background"]
    elif name == 'cityscapes_fine_sem_seg_val':
        class_names = CITYSCAPES
    elif name == 'cityscapes_fine_instance_seg_val':
        class_names = CITYSCAPES_THING + ["background"]
    elif name in ['cityscapes_fine_panoptic_val', 'cityscapes_fine_panoptic_train']:
        class_names = CITYSCAPES + ["background"]
    elif name == 'bdd10k_val_sem_seg':
        class_names = BDD_SEM
    elif name == 'bdd10k_40_panoptic_val':