smooth sequences

1 files changed, 2 insertions, 2 deletions

diff --git a/run.py b/run.py
index 6563b43..8490a57 100644
--- a/run.py
+++ b/run.py
@@ -119,8 +119,8 @@ def average_two_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, st
 	frames = []
 	steps *= dilate
 	for i in range(0, steps * dilate, dilate):
-		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)))
+		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))
 		a_np = load_image(a_fn)
 		b_np = load_image(b_fn)