From ebbb358b542ee4a771879f31741c69b1fdec0cd7 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 25 Jun 2018 18:08:04 +0200 Subject: process sequences --- run.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'run.py') diff --git a/run.py b/run.py index addb24c..be54479 100644 --- a/run.py +++ b/run.py @@ -85,9 +85,9 @@ def recurse_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, count, if (index % 10) == 0: print("{}...".format(index)) step /= 2 - a_img = load_image(os.path.join(a, "frame_{:0>5}.png".format(int(frame_index + a_offset)))) - b_img = load_image(os.path.join(b, "frame_{:0>5}.png".format(int(frame_index + b_offset)))) - frame = recurse_two_frames(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, a_img, b_img, frame_index, count, count / 2) + a_np = load_image(os.path.join(a, "frame_{:0>5}.png".format(int(frame_index + a_offset)))) + b_np = load_image(os.path.join(b, "frame_{:0>5}.png".format(int(frame_index + b_offset)))) + frame = recurse_two_frames(moduleNetwork, tensorOutput, a_np, b_np, frame_index, count, count / 2) if step < 2: return [frame] else: -- cgit v1.2.3-70-g09d2