diff options
Diffstat (limited to 'run.py')
| -rw-r--r-- | run.py | 3 |
1 files changed, 2 insertions, 1 deletions
@@ -159,6 +159,7 @@ def smooth_frames(moduleNetwork, tensorOutput, frames, smooth): return frames print("smoothing every other frame") firstFrame = frames[0] + nextFrame = None new_frames = [firstFrame] for i in range(1, len(frames)-2, 2): firstFrame = frames[i] @@ -171,7 +172,7 @@ def smooth_frames(moduleNetwork, tensorOutput, frames, smooth): new_frames += [firstFrame] new_frames += [middle_np] - if nextFrame: + if nextFrame is not None: new_frames += [nextFrame] new_frames += [frames[len(frames)-1]] return new_frames |