diff options
Diffstat (limited to 'live-mogrify.py')
| -rw-r--r-- | live-mogrify.py | 9 |
1 files changed, 0 insertions, 9 deletions
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() |