diff --git a/.gitignore b/.gitignore index 67bfd9b7..ea1b5576 100644 --- a/.gitignore +++ b/.gitignore @@ -10,4 +10,3 @@ tmp/ tmp/*.json .DS_Store .vscode/ -tmp/ diff --git a/data/stable_jerry.mp4 b/data/stable_jerry.mp4 new file mode 100644 index 00000000..1428678f Binary files /dev/null and b/data/stable_jerry.mp4 differ diff --git a/src/main.py b/src/main.py index 4d4c5eba..25f52b30 100644 --- a/src/main.py +++ b/src/main.py @@ -40,15 +40,21 @@ def main(video_path): modelrunner.run() modelrunner.pose() people_output, ball_output, pose_output = modelrunner.fetch_output() - output_video_path = 'tmp/court_video.mp4' - output_video_path_reenc = 'tmp/court_video_reenc.mp4' + output_video_path = "tmp/court_video.mp4" + output_video_path_reenc = "tmp/court_video_reenc.mp4" + + processrunner = ProcessRunner( + video_path, + people_output, + ball_output, + output_video_path, + output_video_path_reenc, + ) - processrunner = ProcessRunner(video_path, people_output, ball_output, output_video_path, - output_video_path_reenc) processrunner.run() results = processrunner.get_results() return results - + if __name__ == "__main__": import sys