endswith

1 files 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)