diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-04-18 12:57:43 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-04-18 12:57:43 +0200 |
| commit | eb8e0d995e30b47486b0b5efdec009a3171fe244 (patch) | |
| tree | 51305f31e96e6e3fb2df4f08eabe35afe85ab6cc | |
| parent | a4ed9271dea2f9e8b7849b3a1227216ccc3fb58b (diff) | |
k
| -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 \ |