summaryrefslogtreecommitdiff
path: root/run.py
diff options
context:
space:
mode:
Diffstat (limited to 'run.py')
-rw-r--r--run.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/run.py b/run.py
index 2551b5d..db015c8 100644
--- a/run.py
+++ b/run.py
@@ -99,11 +99,11 @@ 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
- a_fn = os.path.join(a, "frame_{:0>5}.png".format(int(steps + a_offset)))
+ a_fn = os.path.join(a, "frame_{:0>5}.png".format(int(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))))
+ b_fn = os.path.join(b, "frame_{:0>5}.png".format(int(b_offset + len(frames))))
if os.path.exists(b_fn):
frames += [ load_image(b_fn) ]
return frames