solve the bug when seg_prefix is list (#1906)

537bb34c · saltknox · Kai Chen · b9acff16 · 537bb34c
Commit 537bb34c authored Jan 02, 2020 by saltknox Committed by Kai Chen Jan 02, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmdet/datasets/builder.py mmdet/datasets/builder.py +1 -1

No files found.
--- a/mmdet/datasets/builder.py
+++ b/mmdet/datasets/builder.py
@@ -8,7 +8,7 @@ from .registry import DATASETS
 def _concat_dataset(cfg, default_args=None):
    ann_files = cfg['ann_file']
    img_prefixes = cfg.get('img_prefix', None)
-    seg_prefixes = cfg.get('seg_prefixes', None)
+    seg_prefixes = cfg.get('seg_prefix', None)
    proposal_files = cfg.get('proposal_file', None)

    datasets = []