def register_coco_panoptic_separated(
name,
metadata,
image_root,
panoptic_root,
panoptic_json,
sem_seg_root,
instances_json
):
panoptic_name = name + "_separated"
DatasetCatalog.register(
panoptic_name,
lambda: merge_to_panoptic(
load_coco_json(instances_json, image_root, panoptic_name),
load_sem_seg(sem_seg_root, image_root),
),
)
MetadataCatalog.get(panoptic_name).set(
panoptic_root=panoptic_root,
image_root=image_root,
panoptic_json=panoptic_json,
sem_seg_root=sem_seg_root,
json_file=instances_json,
evaluator_type="coco_panoptic_seg",
**metadata
)
semantic_name = name + "_stuffonly"
DatasetCatalog.register(semantic_name, lambda: load_sem_seg(sem_seg_root, image_root))
MetadataCatalog.get(semantic_name).set(
sem_seg_root=sem_seg_root,
image_root=image_root,
evaluator_type="sem_seg",
**metadata
)
def register_all_cityscapes(root):
for key, (image_dir, gt_dir) in _RAW_CITYSCAPES_SPLITS.items():
meta = _get_builtin_metadata("cityscapes")
image_dir = os.path.join(root, image_dir)
gt_dir = os.path.join(root, gt_dir)
inst_key = key.format(task="instance_seg")
DatasetCatalog.register(
inst_key,
lambda x=image_dir, y=gt_dir: load_cityscapes_instances(
x, y, from_json=True, to_polygons=True),
)
MetadataCatalog.get(inst_key).set(image_dir=image_dir,
gt_dir=gt_dir,
evaluator_type="cityscapes_instance",
**meta)
sem_key = key.format(task="sem_seg")
DatasetCatalog.register(
sem_key,
lambda x=image_dir, y=gt_dir: load_cityscapes_semantic(x, y))
MetadataCatalog.get(sem_key).set(image_dir=image_dir,
gt_dir=gt_dir,
evaluator_type="cityscapes_sem_seg",
**meta)
def register_lvis_instances(name, metadata, json_file, image_root):
DatasetCatalog.register(name, lambda: load_lvis_json(json_file, image_root, name))
MetadataCatalog.get(name).set(
json_file=json_file,
image_root=image_root,
evaluator_type="lvis",
**metadata
)
def register_visdrone(name, dirname, split):
DatasetCatalog.register(name, lambda: load_vis_instances(dirname, split))
MetadataCatalog.get(name).set(thing_classes=CLASS_NAMES,
dirname=dirname,
split=split)