endswith

1 files changed, 1 insertions, 1 deletions

diff --git a/run.py b/run.py
index 36364ea..33158f0 100644
--- a/run.py
+++ b/run.py
@@ -103,7 +103,7 @@ def process_two_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, st
   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(b_offset + len(frames)/2)))
+  b_fn = os.path.join(b, "frame_{:0>5}.png".format(int(b_offset + len(frames) - dilate)))
   if os.path.exists(b_fn):
     frames += [ load_image(b_fn) ]
   return frames