diff options
| -rw-r--r-- | run.py | 9 |
1 files changed, 8 insertions, 1 deletions
@@ -99,7 +99,14 @@ def recurse_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, count, def process_two_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, steps, dilate): steps *= dilate - return recurse_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, steps, steps, steps/2, dilate) + a_fn = os.path.join(a, "frame_{:0>5}.png".format(int(steps + a_offset))) + frames = [] + frames += load_image(a_fn) + frames += recurse_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, steps, steps, steps/2, dilate) + b_fn = os.path.join(b, "frame_{:0>5}.png".format(int(steps + b_offset + len(frames)))) + if os.path.exists(b_fn): + frames += load_image(b_fn) + return frames def average_two_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, steps, dilate): global index |