diff options
| -rw-r--r-- | data/recursive_dataset.py | 2 | ||||
| -rw-r--r-- | live-mogrify.py | 9 |
2 files changed, 0 insertions, 11 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 1d78877..1d5207b 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -19,9 +19,7 @@ class RecursiveDataset(BaseDataset): def __getitem__(self, index): A_path = os.path.join(self.opt.render_dir, "frame_{:05d}.png".format(index)) - print(A_path) while not os.path.exists(A_path): - print("geevent sleep? {}".format(self.opt.poll_delay)) time.sleep(self.opt.poll_delay) A_img = Image.open(A_path).convert('RGB') A = self.transform(A_img) diff --git a/live-mogrify.py b/live-mogrify.py index a1836cc..97e7aeb 100644 --- a/live-mogrify.py +++ b/live-mogrify.py @@ -140,18 +140,14 @@ class Listener(): def process_live_input(opt, data_opt, rpc_client): A_offset, A_im, A_dir = load_first_frame(opt, data_opt) - print("load first frame") data_loader = CreateRecursiveDataLoader(opt) - print("made data loader") dataset = data_loader.load_data() - print("load data") model = create_model(opt) print("generating...") last_im = None for i, data in enumerate(data_loader): - print("{}...".format(i)) if i >= opt.how_many: break model.set_input(data) @@ -212,10 +208,6 @@ def process_live_input(opt, data_opt, rpc_client): last_im = im.copy().astype('uint8') next_im = im - # image_pil = Image.fromarray(next_im, mode='RGB') - # im = np.asarray(image_pil).astype('uint8') - #print(im.shape, im.dtype) - next_img = process_image(opt, data_opt, next_im) if data_opt.send_image == 'sequence': @@ -230,7 +222,6 @@ def process_live_input(opt, data_opt, rpc_client): cv2.imwrite(tmp_path, next_img) os.rename(tmp_path, next_path) - print('wrote {}'.format(next_path)) if __name__ == '__main__': listener = Listener() |