diff options
| -rw-r--r-- | data/recursive_dataset.py | 2 | ||||
| -rwxr-xr-x | recursive.sh | 4 | ||||
| -rw-r--r-- | test.py | 1 |
3 files changed, 4 insertions, 3 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index eea4ea8..998a96c 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 + self.next_image = opt.dataroot + 'frame_0001.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 3f71494..8cbbf36 100755 --- a/recursive.sh +++ b/recursive.sh @@ -1,9 +1,9 @@ python test.py \ --dataroot /home/lens/Desktop/thumbs/woodscaled_4/A/ \ --name woodscaled_4_pix2pix \ - --how_many 1 \ + --how_many 2 \ --model test \ --which_model_netG unet_256 \ --which_direction AtoB \ - --dataset_mode aligned \ + --dataset_mode recursive \ --norm batch @@ -31,6 +31,7 @@ if __name__ == '__main__': print('%04d: process image... %s' % (i, img_path)) ims = visualizer.save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio) print(ims) + print dataset.name() if dataset.name() == 'RecursiveImageDataset': # dataset.append(ims) print(ims) |