From 5c2e68bb4c20573e8144d92b989454efe886fbbb Mon Sep 17 00:00:00 2001
From: simon wu <wswday@sina.com>
Date: Tue, 6 Aug 2019 19:57:30 +0800
Subject: [PATCH] Support different type of dataset for training (#1133)
---
mmdet/datasets/builder.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/mmdet/datasets/builder.py b/mmdet/datasets/builder.py
index a8b0e29..1d101ff 100644
--- 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)):
--
GitLab