diff options
| -rw-r--r-- | data/recursive_dataset.py | 2 | ||||
| -rwxr-xr-x | recursive.sh | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 998a96c..1eccef7 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -8,7 +8,7 @@ class RecursiveDataset(BaseDataset): def initialize(self, opt): self.opt = opt self.root = opt.dataroot - self.next_image = opt.dataroot + 'frame_0001.png' + self.next_image = opt.dataroot + 'frame_0000.png' self.dir_A = os.path.join(opt.dataroot) self.A_paths = make_dataset(self.dir_A) self.A_paths = sorted(self.A_paths) diff --git a/recursive.sh b/recursive.sh index 8cbbf36..3cff152 100755 --- a/recursive.sh +++ b/recursive.sh @@ -1,5 +1,5 @@ python test.py \ - --dataroot /home/lens/Desktop/thumbs/woodscaled_4/A/ \ + --dataroot /home/lens/Desktop/thumbs/woodscaled_4/A/train/ \ --name woodscaled_4_pix2pix \ --how_many 2 \ --model test \ |