debuggin..

1 files changed, 3 insertions, 9 deletions

diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py
index 828adb5..863c385 100644
--- a/data/recursive_dataset.py
+++ b/data/recursive_dataset.py
@@ -21,15 +21,9 @@ class RecursiveDataset(BaseDataset):
         A_path = os.path.join(self.opt.render_dir, "frame_{:05d}.png".format(index))
         # print(A_path)
         if not os.path.exists(A_path):
-            A_path2 = os.path.join(self.opt.render_dir, "frame_{:05d}.png".format(index+1))
-            if os.path.exists(A_path2):
-                # print('oops: {}'.format(A_path2))
-                A_path = A_path2
-            else:
-                # print('not there yet... {}'.format(A_path))
-                while not os.path.exists(A_path):
-                    # print('sleeping for {}'.format(self.opt.poll_delay))
-                    time.sleep(self.opt.poll_delay)
+            while not os.path.exists(A_path):
+                # print('sleeping for {}'.format(self.opt.poll_delay))
+                time.sleep(self.opt.poll_delay)
         # print("got {}".format(A_path))
         A_img = Image.open(A_path).convert('RGB')
         A = self.transform(A_img)