diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-04-18 12:35:39 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-04-18 12:35:39 +0200 |
| commit | d3044d50514586b0cf9702bc2c16a6486af249f6 (patch) | |
| tree | 2c05c6d37efe7b23abc93c5419d6b2cfb037e6bb /data/__init__.py | |
| parent | e3726af25b83134d6240b926386fa0243f6a6a96 (diff) | |
setting up recursive dataset
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) |