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