diff options
Diffstat (limited to 'data/recursive_dataset.py')
| -rw-r--r-- | data/recursive_dataset.py | 6 |
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) |