summaryrefslogtreecommitdiff
path: root/data/recursive_dataset.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-04-18 12:57:43 +0200
committerJules Laplace <julescarbon@gmail.com>2018-04-18 12:57:43 +0200
commiteb8e0d995e30b47486b0b5efdec009a3171fe244 (patch)
tree51305f31e96e6e3fb2df4f08eabe35afe85ab6cc /data/recursive_dataset.py
parenta4ed9271dea2f9e8b7849b3a1227216ccc3fb58b (diff)
k
Diffstat (limited to 'data/recursive_dataset.py')
-rw-r--r--data/recursive_dataset.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py
index c95654d..eea4ea8 100644
--- a/data/recursive_dataset.py
+++ b/data/recursive_dataset.py
@@ -9,9 +9,9 @@ class RecursiveDataset(BaseDataset):
self.opt = opt
self.root = opt.dataroot
self.next_image = opt.dataroot
- #self.dir_A = os.path.join(opt.dataroot)
- #self.A_paths = make_dataset(self.dir_A)
- #self.A_paths = sorted(self.A_paths)
+ self.dir_A = os.path.join(opt.dataroot)
+ self.A_paths = make_dataset(self.dir_A)
+ self.A_paths = sorted(self.A_paths)
self.transform = get_transform(opt)