diff options
| -rw-r--r-- | data/recursive_dataset.py | 6 | ||||
| -rwxr-xr-x | 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 \ |