diff options
Diffstat (limited to 'run.py')
| -rw-r--r-- | run.py | 6 |
1 files changed, 4 insertions, 2 deletions
@@ -69,11 +69,13 @@ 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) + print("next index: {} - {}".format(next_index, step)) + next_index = frame_index - step next_a_np = a_np next_b_np = middle_np else: - next_index = frame_index + (step) + print("next index: {} + {}".format(next_index, step)) + next_index = frame_index + step next_a_np = middle_np next_b_np = b_np return recurse_two_frames(moduleNetwork, tensorOutput, next_a_np, next_b_np, frame_index, next_index, step/2, depth+1) |