From 555bd36b2da2341dd506ba8ecbf48a248c924ce4 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 21 May 2018 18:53:16 +0200 Subject: feel better... with metta --- live-mogrify.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'live-mogrify.py') diff --git a/live-mogrify.py b/live-mogrify.py index bf3975a..b511af1 100644 --- a/live-mogrify.py +++ b/live-mogrify.py @@ -273,6 +273,7 @@ def process_live_input(opt, data_opt, rpc_client, model): tmp_path = opt.render_dir + "frame_{:05d}_tmp.png".format(i+1) next_path = opt.render_dir + "frame_{:05d}.png".format(i+1) current_path = opt.render_dir + "ren_{:05d}.png".format(i+1) + meta = { 'i': i, 'sequence_i': sequence_i, 'sequence_len': len(sequence) } if data_opt.sequence and len(sequence): sequence_path = sequence[sequence_i] if sequence_i >= len(sequence)-1: @@ -283,7 +284,7 @@ def process_live_input(opt, data_opt, rpc_client, model): if data_opt.send_image == 'b': image_pil = Image.fromarray(im, mode='RGB') - rpc_client.send_pil_image("frame_{:05d}.png".format(i+1), image_pil) + rpc_client.send_pil_image("frame_{:05d}.png".format(i+1), meta, image_pil) if data_opt.store_a is not True: os.remove(last_path) @@ -333,14 +334,14 @@ def process_live_input(opt, data_opt, rpc_client, model): next_img = process_image(opt, data_opt, next_im) if data_opt.send_image == 'sequence': - rpc_client.send_pil_image("frame_{:05d}.png".format(i+1), A_img) + rpc_client.send_pil_image("frame_{:05d}.png".format(i+1), meta, A_img) if data_opt.send_image == 'recursive': pil_im = Image.fromarray(next_im) - rpc_client.send_pil_image("frame_{:05d}.png".format(i+1), pil_im) + rpc_client.send_pil_image("frame_{:05d}.png".format(i+1), meta, pil_im) if data_opt.send_image == 'a': rgb_im = cv2.cvtColor(next_img, cv2.COLOR_BGR2RGB) pil_im = Image.fromarray(rgb_im) - rpc_client.send_pil_image("frame_{:05d}.png".format(i+1), pil_im) + rpc_client.send_pil_image("frame_{:05d}.png".format(i+1), meta, pil_im) cv2.imwrite(tmp_path, next_img) os.rename(tmp_path, next_path) -- cgit v1.2.3-70-g09d2