From 3936a9968558681aff5028d28bb25dcf343c7ed9 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 26 Jun 2018 14:19:09 +0200 Subject: dataset --- run.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'run.py') 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", -- cgit v1.2.3-70-g09d2