Commit 58f0ae3d authored by Yanghan Wang's avatar Yanghan Wang Committed by Facebook GitHub Bot
Browse files

enable hive loader for person segmentation

Reviewed By: leitian

Differential Revision: D28363172

fbshipit-source-id: e69a71e6525dc9b76171b0cdc5f55ee8d188d6cc
parent de0829f1
...@@ -7,6 +7,7 @@ import importlib ...@@ -7,6 +7,7 @@ import importlib
import logging import logging
import os import os
from d2go.data.fb.adhoc_hive_dataset import COCOHiveLoaderConfig
from d2go.utils.helper import get_dir_path from d2go.utils.helper import get_dir_path
from detectron2.data import DatasetCatalog, MetadataCatalog from detectron2.data import DatasetCatalog, MetadataCatalog
...@@ -47,6 +48,7 @@ def _register_extended_coco(dataset_name, split_dict): ...@@ -47,6 +48,7 @@ def _register_extended_coco(dataset_name, split_dict):
meta_data = split_dict.get("meta_data", {}) meta_data = split_dict.get("meta_data", {})
MetadataCatalog.get(dataset_name).set( MetadataCatalog.get(dataset_name).set(
evaluator_type=evaluator_type, evaluator_type=evaluator_type,
auto_hive_loader_config=COCOHiveLoaderConfig,
json_file=json_file, json_file=json_file,
image_root=image_root, image_root=image_root,
**meta_data **meta_data
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment