diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-25 18:08:55 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-25 18:08:55 +0200 |
| commit | 1aea4a9d637fd570665d7bab4fb11b28c5c5a0f6 (patch) | |
| tree | 26bc53794719230e56f795903d05ce0d61f94bf9 /run.py | |
| parent | ebbb358b542ee4a771879f31741c69b1fdec0cd7 (diff) | |
process sequences
Diffstat (limited to 'run.py')
| -rw-r--r-- | run.py | 7 |
1 files changed, 5 insertions, 2 deletions
@@ -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] |
