From d3044d50514586b0cf9702bc2c16a6486af249f6 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 18 Apr 2018 12:35:39 +0200 Subject: setting up recursive dataset --- data/__init__.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'data/__init__.py') 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) -- cgit v1.2.3-70-g09d2