diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-25 18:06:56 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-25 18:06:56 +0200 |
| commit | 579779131220af82f9cc155331e42803a6f0807b (patch) | |
| tree | 58cd007828ed733c216ce01a34ce8bf2d95bb6db | |
| parent | df32a7acda9cada817dcb7730b2d7d4335646c51 (diff) | |
process sequences
| -rw-r--r-- | run.py | 6 |
1 files changed, 3 insertions, 3 deletions
@@ -87,12 +87,12 @@ def recurse_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, count, 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 = process_two_frames(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, a_img, b_img, frame_index, count, count / 2) + frame = recurse_two_frames(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, a_img, b_img, frame_index, count, count / 2) if step < 2: return [frame] else: - left = process_two_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, count, step, frame_index - (step/2)) - right = process_two_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, count, step, frame_index + (step/2)) + left = recurse_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, count, step, frame_index - (step/2)) + right = recurse_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, count, step, frame_index + (step/2)) return left + [frame] + right def process_two_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, steps): |