dataset

1 files changed, 2 insertions, 1 deletions

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