debuggin..

2 files changed, 4 insertions, 11 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)
diff --git a/live-mogrify.py b/live-mogrify.py
index 1d8126a..3a66d39 100644
--- a/live-mogrify.py
+++ b/live-mogrify.py
@@ -8,11 +8,10 @@ from models import create_model
 # from util.visualizer import Visualizer
 from util.util import mkdirs, crop_image
 from util import html
-from shutil import move, copyfile
+from shutil import move, copyfile, rmtree
 from PIL import Image, ImageOps
 from skimage.transform import resize
 from scipy.misc import imresize
-from shutil import copyfile, rmtree
 import numpy as np
 import cv2
 from datetime import datetime