Spaces:
Runtime error
Runtime error
| import os | |
| from detectron2.data import DatasetCatalog, MetadataCatalog | |
| from detectron2.data.datasets import load_sem_seg | |
| import copy | |
| def _get_pascal_voc_meta(): | |
| voc_classes = ["aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat", "chair", "cow", "diningtable", "dog", "horse", "motorbike", "person", "pottedplant", "sheep", "sofa", "train", "tvmonitor"] | |
| voc_colors = [[0, 0, 0], [128, 0, 0], [0, 128, 0], [128, 128, 0], [0, 0, 128], | |
| [128, 0, 128], [0, 128, 128], [128, 128, 128], [64, 0, 0], | |
| [192, 0, 0], [64, 128, 0], [192, 128, 0], [64, 0, 128], | |
| [192, 0, 128], [64, 128, 128], [192, 128, 128], [0, 64, 0], | |
| [128, 64, 0], [0, 192, 0], [128, 192, 0], [0, 64, 128]] | |
| ret = { | |
| "stuff_classes" : voc_classes, | |
| "stuff_colors" : voc_colors, | |
| } | |
| return ret | |
| def register_all_pascal_voc(root): | |
| root = os.path.join(root, "VOCdevkit/VOC2012") | |
| meta = _get_pascal_voc_meta() | |
| for name, image_dirname, sem_seg_dirname in [ | |
| ("test", "JPEGImages", "annotations_detectron2"), | |
| ("test_background", "JPEGImages", "annotations_detectron2_bg"), | |
| ]: | |
| image_dir = os.path.join(root, image_dirname) | |
| gt_dir = os.path.join(root, sem_seg_dirname, 'val') | |
| name = f"voc_2012_{name}_sem_seg" | |
| DatasetCatalog.register(name, lambda x=image_dir, y=gt_dir: load_sem_seg(y, x, gt_ext='png', image_ext='jpg')) | |
| if "background" in name: | |
| MetadataCatalog.get(name).set(image_root=image_dir, seg_seg_root=gt_dir, evaluator_type="sem_seg_background", ignore_label=255, | |
| stuff_classes=meta["stuff_classes"] + ["background"], stuff_colors=meta["stuff_colors"]) | |
| else: | |
| MetadataCatalog.get(name).set(image_root=image_dir, seg_seg_root=gt_dir, evaluator_type="sem_seg", ignore_label=255, **meta,) | |
| def register_all_pascal_voc_background(root): | |
| root = os.path.join(root, "VOCdevkit/VOC2012") | |
| meta = _get_pascal_voc_meta() | |
| meta["stuff_classes"] = meta["stuff_classes"] + ["background"] | |
| for name, image_dirname, sem_seg_dirname in [ | |
| ("test_background", "image", "label_openseg_background20"), | |
| ]: | |
| image_dir = os.path.join(root, image_dirname, 'validation') | |
| gt_dir = os.path.join(root, sem_seg_dirname, 'validation') | |
| name = f"voc_2012_{name}_sem_seg" | |
| DatasetCatalog.register(name, lambda x=image_dir, y=gt_dir: load_sem_seg(y, x, gt_ext='png', image_ext='jpg')) | |
| MetadataCatalog.get(name).set(image_root=image_dir, seg_seg_root=gt_dir, evaluator_type="sem_seg_background", ignore_label=255, **meta,) | |
| _root = os.getenv("DETECTRON2_DATASETS", "datasets") | |
| register_all_pascal_voc(_root) | |
| #register_all_pascal_voc_background(_root) |