more general

4 files changed, 7 insertions, 1 deletions

diff --git a/.gitignore b/.gitignore
index 4fdef3e..7290625 100644
--- a/.gitignore
+++ b/.gitignore
@@ -41,3 +41,4 @@ test/.coverage
 test/data/legacy_serialized.pt
 *~
 .idea
+recursive/
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py
index 4bd0798..f8a5da8 100644
--- a/data/recursive_dataset.py
+++ b/data/recursive_dataset.py
@@ -1,3 +1,5 @@
+import os
+import time
 import os.path
 from data.base_dataset import BaseDataset, get_transform
 from data.image_folder import make_dataset
@@ -37,6 +39,8 @@ class RecursiveDataset(BaseDataset):
     def __getitem__(self, index):
         A_path = self.opt.results_dir + "/" + self.opt.name + "/exp:" + self.opt.experiment + "/frame_{:05d}.png".format(index)
         print('next image: ' + A_path)
+        while not os.path.exists(A_path):
+            time.sleep(0.1)
         A_img = Image.open(A_path).convert('RGB')
         A = self.transform(A_img)
         if self.opt.which_direction == 'BtoA':
diff --git a/recursive.sh b/recursive.sh
index 3cff152..a3856ab 100755
--- a/recursive.sh
+++ b/recursive.sh
@@ -1,7 +1,7 @@
 python test.py \
   --dataroot /home/lens/Desktop/thumbs/woodscaled_4/A/train/ \
   --name woodscaled_4_pix2pix \
-  --how_many 2 \
+  --how_many 2000 \
   --model test  \
   --which_model_netG unet_256  \
   --which_direction AtoB  \
diff --git a/test.py b/test.py
index f5e9dd7..8409bda 100644
--- a/test.py
+++ b/test.py
@@ -7,6 +7,7 @@ from util import html, mkdir
 from shutil import movefile
 
 import subprocess
+from time import sleep
 
 
 if __name__ == '__main__':