dataset

1 files changed, 2 insertions, 2 deletions

diff --git a/run.py b/run.py
index 80eaecc..ed036a0 100644
--- a/run.py
+++ b/run.py
@@ -188,7 +188,7 @@ def dilate_frames(moduleNetwork, tensorOutput, frames, dilate):
 	new_frames += [nextFrame]
 	return new_frames
 
-def store_frames(frames, outputPath, inputFirst=None, inputSecond=None, endpoint=None):
+def store_frames(frames, outputPath, inputFirst=None, inputSecond=None, endpoint=None, dataset="unknown"):
 	print('writing {}'.format(outputPath))
 	print('frames: {}'.format(len(frames)))
 	writer = FFMPEG_VideoWriter(outputPath, (1024, 512), 25)
@@ -199,7 +199,7 @@ def store_frames(frames, outputPath, inputFirst=None, inputSecond=None, endpoint
 	if inputSecond is not None:
 		writer.write_frame(inputSecond)
 	writer.close()
-	
+
 	if endpoint is not None:
 	  os.system(["curl",
 	    "-X", "POST",