diff options
Diffstat (limited to 'run.py')
| -rw-r--r-- | run.py | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -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", |