dataset

1 files changed, 2 insertions, 1 deletions

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