summaryrefslogtreecommitdiff
path: root/run.py
diff options
context:
space:
mode:
Diffstat (limited to 'run.py')
-rw-r--r--run.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/run.py b/run.py
index d59a13b..d004ed0 100644
--- a/run.py
+++ b/run.py
@@ -139,7 +139,8 @@ def smooth_frames(moduleNetwork, tensorOutput, frames, smooth):
process(moduleNetwork, tensorInputFirst, tensorInputSecond, tensorOutput)
middle_np = tensorOutput.clamp(0.0, 1.0).numpy()
- new_frames += [firstFrame, middle_np]
+ new_frames += [firstFrame]
+ new_frames += [middle_np]
new_frames += [nextFrame]
return new_frames