slight alteration

CornellDataScience · Oct 21, 2023 · 4f9b2b5 · 4f9b2b5
1 parent be4cd9f
commit 4f9b2b5
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git 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 → src/pose_estimation/best.pt b/best.pt → src/pose_estimation/best.pt
diff --git 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