From 8de246ec1de1107c7510b45ad34685899d2e9bb8 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 26 Jun 2018 16:13:35 +0200 Subject: dataset --- run.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/run.py b/run.py index ec96399..971a779 100644 --- a/run.py +++ b/run.py @@ -171,7 +171,8 @@ def smooth_frames(moduleNetwork, tensorOutput, frames, smooth): new_frames += [firstFrame] new_frames += [middle_np] - new_frames += [nextFrame] + if nextFrame: + new_frames += [nextFrame] new_frames += [frames[len(frames)-1]] return new_frames -- cgit v1.2.3-70-g09d2