diff --git a/src/modelrunner.py b/src/modelrunner.py index 7156831f..5898a8fd 100644 --- a/src/modelrunner.py +++ b/src/modelrunner.py @@ -55,7 +55,7 @@ def run(self): self.output_dict = pickle.load(f) def pose(self): - model = YOLO('best.pt') + model = YOLO('src/pose_estimation/best.pt') results = model( source = self.video_path, show=False, diff --git a/best.pt b/src/pose_estimation/best.pt similarity index 99% rename from best.pt rename to src/pose_estimation/best.pt index 02924aca..33342b1f 100644 Binary files a/best.pt and b/src/pose_estimation/best.pt differ diff --git a/src/pose_estimation/pose_estimate.py b/src/pose_estimation/pose_estimate.py index 20be080f..ddf74774 100644 --- a/src/pose_estimation/pose_estimate.py +++ b/src/pose_estimation/pose_estimate.py @@ -4,7 +4,7 @@ from ultralytics import YOLO class PoseEstimator: - def __init__(self, model_path='best.pt', video_path='res/pose_results/test_multiple_people.mp4', combinations=None): + def __init__(self, model_path='src/pose_estimation/best.pt', video_path='res/pose_results/test_multiple_people.mp4', combinations=None): # Initialize paths, model, and combinations of keypoints to calculate angles self.model_path = model_path self.video_path = video_path