From 6e13e93c81369507f089b753c72e4aba7be3a58b Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Wed, 18 Apr 2018 13:11:39 +0200
Subject: k

---
 data/recursive_dataset.py | 2 +-
 recursive.sh              | 4 ++--
 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
diff --git a/test.py b/test.py
index 2263122..09ae7dc 100644
--- a/test.py
+++ b/test.py
@@ -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)
-- 
cgit v1.2.3-70-g09d2