summaryrefslogtreecommitdiff
path: root/run.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-06-25 18:22:39 +0200
committerJules Laplace <julescarbon@gmail.com>2018-06-25 18:22:39 +0200
commitd8f4d9eda95775506f35ba87f2706a1fb3d61936 (patch)
tree3498b9bbcfeb8afc93a4c636a6b392b63bea44cc /run.py
parent32c9dfad1d5e328325f89ba187cd3fbc2e2843b7 (diff)
process sequences
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 f87dee7..61f767f 100644
--- a/run.py
+++ b/run.py
@@ -69,12 +69,12 @@ def recurse_two_frames(moduleNetwork, tensorOutput, a_np, b_np, frame_index, mor
middle_img = (numpy.rollaxis(middle_np, 0, 3)[:,:,::-1] * 255.0).astype(numpy.uint8)
return middle_img
if morph_index > frame_index:
- next_index = frame_index - step
+ next_index = morph_index - step
next_a_np = a_np
next_b_np = middle_np
print("next index: {} - {}".format(next_index, step))
else:
- next_index = frame_index + step
+ next_index = morph_index + step
next_a_np = middle_np
next_b_np = b_np
print("next index: {} + {}".format(next_index, step))