summaryrefslogtreecommitdiff
path: root/run.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-06-25 18:08:55 +0200
committerJules Laplace <julescarbon@gmail.com>2018-06-25 18:08:55 +0200
commit1aea4a9d637fd570665d7bab4fb11b28c5c5a0f6 (patch)
tree26bc53794719230e56f795903d05ce0d61f94bf9 /run.py
parentebbb358b542ee4a771879f31741c69b1fdec0cd7 (diff)
process sequences
Diffstat (limited to 'run.py')
-rw-r--r--run.py7
1 files changed, 5 insertions, 2 deletions
diff --git a/run.py b/run.py
index be54479..39aeab3 100644
--- a/run.py
+++ b/run.py
@@ -85,8 +85,11 @@ def recurse_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, count,
if (index % 10) == 0:
print("{}...".format(index))
step /= 2
- a_np = load_image(os.path.join(a, "frame_{:0>5}.png".format(int(frame_index + a_offset))))
- b_np = load_image(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(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))
+ a_np = load_image(a_fn)
+ b_np = load_image(b_fn)
frame = recurse_two_frames(moduleNetwork, tensorOutput, a_np, b_np, frame_index, count, count / 2)
if step < 2:
return [frame]