diff options
| -rw-r--r-- | data/recursive_dataset.py | 12 | ||||
| -rw-r--r-- | live-mogrify.py | 3 |
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 |