From eb8e0d995e30b47486b0b5efdec009a3171fe244 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 18 Apr 2018 12:57:43 +0200 Subject: k --- data/recursive_dataset.py | 6 +++--- recursive.sh | 2 +- 2 files changed, 4 insertions(+), 4 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) diff --git a/recursive.sh b/recursive.sh index adcb1d1..3f71494 100755 --- a/recursive.sh +++ b/recursive.sh @@ -1,5 +1,5 @@ python test.py \ - --dataroot /home/lens/Desktop/thumbs/woodscaled_4/A/train/frame_0000.png \ + --dataroot /home/lens/Desktop/thumbs/woodscaled_4/A/ \ --name woodscaled_4_pix2pix \ --how_many 1 \ --model test \ -- cgit v1.2.3-70-g09d2