Unverified Commit 1c72acdd authored by ChaimZhu's avatar ChaimZhu Committed by GitHub
Browse files

[Fix]: fix coco registry problem (#1536)

parent fab3db2e
...@@ -3,6 +3,7 @@ import platform ...@@ -3,6 +3,7 @@ import platform
from mmcv.utils import Registry, build_from_cfg from mmcv.utils import Registry, build_from_cfg
from mmdet.datasets import DATASETS as MMDET_DATASETS
from mmdet.datasets.builder import _concat_dataset from mmdet.datasets.builder import _concat_dataset
if platform.system() != 'Windows': if platform.system() != 'Windows':
...@@ -39,7 +40,8 @@ def build_dataset(cfg, default_args=None): ...@@ -39,7 +40,8 @@ def build_dataset(cfg, default_args=None):
dataset = CBGSDataset(build_dataset(cfg['dataset'], default_args)) dataset = CBGSDataset(build_dataset(cfg['dataset'], default_args))
elif isinstance(cfg.get('ann_file'), (list, tuple)): elif isinstance(cfg.get('ann_file'), (list, tuple)):
dataset = _concat_dataset(cfg, default_args) dataset = _concat_dataset(cfg, default_args)
else: elif cfg['type'] in DATASETS._module_dict.keys():
dataset = build_from_cfg(cfg, DATASETS, default_args) dataset = build_from_cfg(cfg, DATASETS, default_args)
else:
dataset = build_from_cfg(cfg, MMDET_DATASETS, default_args)
return dataset return dataset
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