Spaces:
Runtime error
Runtime error
File size: 2,803 Bytes
f8f62f3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
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) |