diff options
| -rw-r--r-- | data/recursive_dataset.py | 3 | ||||
| -rw-r--r-- | live-mogrify.py | 5 |
2 files changed, 6 insertions, 2 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 863c385..dfa1d20 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -19,8 +19,9 @@ class RecursiveDataset(BaseDataset): def __getitem__(self, index): A_path = os.path.join(self.opt.render_dir, "frame_{:05d}.png".format(index)) - # print(A_path) + print("awaiting {}".format(A_path)) if not os.path.exists(A_path): + # print() while not os.path.exists(A_path): # print('sleeping for {}'.format(self.opt.poll_delay)) time.sleep(self.opt.poll_delay) diff --git a/live-mogrify.py b/live-mogrify.py index 3a66d39..477cde7 100644 --- a/live-mogrify.py +++ b/live-mogrify.py @@ -98,6 +98,9 @@ def process_image(opt, data_opt, im): img = im[:, :, ::-1].copy() processed = False + if data_opt.process_frac == 0: + return img + if data_opt.clahe is True: processed = True lab = cv2.cvtColor(img, cv2.COLOR_BGR2LAB) @@ -120,7 +123,7 @@ def process_image(opt, data_opt, im): img = cv2.Canny(img, data_opt.canny_lo, data_opt.canny_hi) img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR) - if processed is False or data_opt.process_frac == 0: + if processed is False: return img src_img = im[:, :, ::-1].copy() |