diff --git a/.DS_Store b/.DS_Store new file mode 100644 index 00000000..6d2b0a56 Binary files /dev/null and b/.DS_Store differ diff --git a/src/main.py b/src/main.py index 67fbd10d..966ac9ad 100644 --- a/src/main.py +++ b/src/main.py @@ -40,7 +40,7 @@ def main(video_path): 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, + processrunner = ProcessRunner(video_path, people_output, ball_output, output_video_path, output_video_path_reenc) processrunner.run() results = processrunner.get_results() diff --git a/src/processing/pose_estimate.py b/src/processing/pose_estimate.py new file mode 100644 index 00000000..c3da9538 --- /dev/null +++ b/src/processing/pose_estimate.py @@ -0,0 +1,5 @@ +from ultralytics import YOLO + +model = YOLO('yolov8m-pose.pt') + +results = model(source = "test_video.mp4", show = True, conf = 0.3, save = True) \ No newline at end of file