Support different type of dataset for training (#1133)

5c2e68bb · simon wu · Kai Chen · 8f7507eb · 5c2e68bb
Commit 5c2e68bb authored Aug 06, 2019 by simon wu Committed by Kai Chen Aug 06, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

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

No files found.
--- a/mmdet/datasets/builder.py
+++ b/mmdet/datasets/builder.py
@@ -28,7 +28,9 @@ def _concat_dataset(cfg, default_args=None):
 def build_dataset(cfg, default_args=None):
-    if cfg['type'] == 'RepeatDataset':
+    if isinstance(cfg, (list, tuple)):
+        dataset = ConcatDataset([build_dataset(c, default_args) for c in cfg])
+    elif cfg['type'] == 'RepeatDataset':
        dataset = RepeatDataset(
            build_dataset(cfg['dataset'], default_args), cfg['times'])
    elif isinstance(cfg['ann_file'], (list, tuple)):