summaryrefslogtreecommitdiff
path: root/run.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-06-25 19:10:54 +0200
committerJules Laplace <julescarbon@gmail.com>2018-06-25 19:10:54 +0200
commit278296e1110bea0abe49af6087e855c312087d92 (patch)
treec351fd50dbd08a3799b86a2cb2e0ba15a81e4d1a /run.py
parentb06cabfaed85179a5c1f8805f5c0f55b3f916a22 (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 cd19277..e1b6a58 100644
--- a/run.py
+++ b/run.py
@@ -176,7 +176,7 @@ elif arguments_mixVideos:
print("{} => {}".format(arguments_strFirst, arguments_strSecond))
outputPath = './renders/' + arguments_strVideoOut
frames = process_two_videos(moduleNetwork, tensorOutput, arguments_strFirst, arguments_strSecond, arguments_aOffset, arguments_bOffset, arguments_steps, arguments_dilate)
- dilate_frames(moduleNetwork, tensorOutput, frames, arguments_dilate)
+ frames = dilate_frames(moduleNetwork, tensorOutput, frames, arguments_dilate)
store_frames(frames, outputPath)
elif arguments_steps == 0:
# Process image
@@ -191,5 +191,5 @@ else:
inputSecond = load_image(arguments_strSecond)
outputPath = './renders/' + arguments_strVideoOut
frames = process_tree(moduleNetwork, inputFirst, inputSecond, tensorOutput, arguments_steps * arguments_dilate, arguments_dilate)
- dilate_frames(moduleNetwork, tensorOutput, frames, arguments_dilate)
+ frames = dilate_frames(moduleNetwork, tensorOutput, frames, arguments_dilate)
store_frames(frames, outputPath, inputFirst, inputSecond)