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
import logging
import os
from d2go.data.fb.adhoc_hive_dataset import COCOHiveLoaderConfig
from d2go.utils.helper import get_dir_path
from detectron2.data import DatasetCatalog, MetadataCatalog
......@@ -47,6 +48,7 @@ def _register_extended_coco(dataset_name, split_dict):
meta_data = split_dict.get("meta_data", {})
MetadataCatalog.get(dataset_name).set(
evaluator_type=evaluator_type,
auto_hive_loader_config=COCOHiveLoaderConfig,
json_file=json_file,
image_root=image_root,
**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