From dd20c2c92469b5fe38c4896a8ce0cc41148cf6e2 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sun, 1 Jul 2018 13:16:25 +0200 Subject: endswith --- run.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/run.py b/run.py index 487d050..09ca983 100644 --- a/run.py +++ b/run.py @@ -83,7 +83,8 @@ def recurse_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, count, step /= 2 a_fn = os.path.join(a, "frame_{:0>5}.png".format(int(frame_index + a_offset))) b_fn = os.path.join(b, "frame_{:0>5}.png".format(int(frame_index + b_offset))) - print("{} => {}".format(a_fn, b_fn)) + print("{}".format(a_fn)) + print("=> {}".format(b_fn)) if not os.path.exists(a_fn) or not os.path.exists(b_fn): return [ None ] a_np = load_image(a_fn) @@ -110,7 +111,8 @@ def average_two_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, st for i in range(1, steps * dilate + 1, dilate): a_fn = os.path.join(a, "frame_{:0>5}.png".format(int(i + a_offset))) b_fn = os.path.join(b, "frame_{:0>5}.png".format(int(i + b_offset))) - print("{} => {}".format(a_fn, b_fn)) + print("{}".format(a_fn)) + print("=> {}".format(b_fn)) if not os.path.exists(a_fn) and not os.path.exists(b_fn): continue a_np = load_image(a_fn) @@ -211,6 +213,7 @@ def pad_frames(writer, base_path, start, end): for index in range(start, end): fn = os.path.join(base_path, "frame_{:0>5}.png".format(int(index + start))) if os.path.exists(fn): + print(fn) img = PIL.Image.open(fn) writer.write_frame(img) -- cgit v1.2.3-70-g09d2