From a94098de2ff61f38861bd491093aa757114f6274 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 26 Jun 2018 16:17:56 +0200 Subject: dataset --- run.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'run.py') diff --git a/run.py b/run.py index 971a779..00535fa 100644 --- a/run.py +++ b/run.py @@ -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 -- cgit v1.2.3-70-g09d2