endswith

1 files changed, 8 insertions, 1 deletions

diff --git a/run.py b/run.py
index efb31b9..98333f8 100644
--- a/run.py
+++ b/run.py
@@ -99,7 +99,14 @@ def recurse_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, count,
 
 def process_two_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, steps, dilate):
   steps *= dilate
-  return recurse_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, steps, steps, steps/2, dilate)
+  a_fn = os.path.join(a, "frame_{:0>5}.png".format(int(steps + a_offset)))
+  frames = []
+  frames += load_image(a_fn)
+  frames += recurse_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, steps, steps, steps/2, dilate)
+  b_fn = os.path.join(b, "frame_{:0>5}.png".format(int(steps + b_offset + len(frames))))
+  if os.path.exists(b_fn):
+    frames += load_image(b_fn)
+  return frames
 
 def average_two_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, steps, dilate):
   global index