diff options
Diffstat (limited to 'data/__init__.py')
| -rw-r--r-- | data/__init__.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/data/__init__.py b/data/__init__.py index 341281d..a69f374 100644 --- a/data/__init__.py +++ b/data/__init__.py @@ -20,6 +20,9 @@ def CreateDataset(opt): elif opt.dataset_mode == 'single': from data.single_dataset import SingleDataset dataset = SingleDataset() + elif opt.dataset_mode == 'recursive': + from data.recursive_dataset import RecursiveDataset + dataset = RecursiveDataset() else: raise ValueError("Dataset [%s] not recognized." % opt.dataset_mode) |