diff options
| -rw-r--r-- | live-mogrify.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/live-mogrify.py b/live-mogrify.py index 6b90298..8166cda 100644 --- a/live-mogrify.py +++ b/live-mogrify.py @@ -38,11 +38,11 @@ def load_opt(): opt.serial_batches = True # no shuffle opt.no_flip = True # no flip - data_opt.tag = get_tag(data_opt) + data_opt.tag = get_tag(opt, data_opt) opt.render_dir = opt.results_dir + opt.name + "/" + tag + "/" return opt, data_opt -def get_tag(data_opt): +def get_tag(opt, data_opt): if data_opt.tag == '': d = datetime.now() tag = data_opt.tag = "{}_{}_{}".format( @@ -61,7 +61,7 @@ def load_first_frame(opt, data_opt): print("preload {}".format(opt.start_img)) A_img = Image.open(opt.start_img).convert('RGB') A_im = np.asarray(A_img) - A = process_image(A_im) + A = process_image(opt, data_opt, A_im) cv2.imwrite(start_img_path, A) numz = re.findall(r'\d+', os.path.basename(opt.start_img)) @@ -77,7 +77,7 @@ def load_first_frame(opt, data_opt): print("Sequence not found") return A_offset, A_im, A_dir -def process_image(im): +def process_image(opt, data_opt, im): img = im[:, :, ::-1].copy() processed = False @@ -198,7 +198,7 @@ def process_live_input(): # im = np.asarray(image_pil).astype('uint8') #print(im.shape, im.dtype) - next_img = process_image(next_im) + next_img = process_image(opt, data_opt, next_im) if opt.send_image == 'sequence': rpc_client.send_pil_image("frame_{:05d}.png".format(i+1), A_img) |