Merge branch 'ball-202' into pose-estimate

CornellDataScience · Oct 25, 2023 · e5e1445 · e5e1445
2 parents 388cfb0 + 1df5d4a
commit e5e1445
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git 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
diff --git 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