going somewhere.

dodgeballtracker/logic.py

@@ -107,9 +107,12 @@ class DodgeballTracker:
 
     def run_model_on_frame(self, model, frame: VideoFrame):
         model_outputs = model(frame)[0]
+        return model_outputs.boxes.data.cpu().numpy()
+        # return model_outputs.boxes.data.tolist()
+        foo = model_outputs.boxes.data
         for data in model_outputs.boxes.data.tolist():
             detected_obj = DetectedObject(*data)
-            if detected_obj.confidence > 0.5:
+            if detected_obj.score > 0.5:
                 yield detected_obj
 
     def persist_points(self):
@@ -130,7 +133,7 @@ class DodgeballTracker:
         return YOLO("yolo11n.pt", verbose=True)
 
     def update_tracker(self, tracker, detected_objs: list[DetectedObject], frame: VideoFrame) -> None:
-        important_detected_objs = [do for do in detected_objs if do.confidence > 0.5]
+        important_detected_objs = [do for do in detected_objs if do.score > 0.5]
 
     def main(self):
         # define some constants
@@ -141,7 +144,11 @@ class DodgeballTracker:
         for i, (frame, ts) in enumerate(iter_frames(video_cap)):
             # run the YOLO model on the frame
             detected_objs = list(self.run_model_on_frame(model, frame))
-            self.video_writer.plot_boxes_to_frame(detected_objs, frame)
+            data_ = tracker.update(
+                np.array(detected_objs), frame)
+            if i == 100:
+                print(data_)
+            # self.video_writer.plot_boxes_to_frame(detected_objs, frame)
             # data_ = tracker.update(np.array([dobj.as_nparray() for dobj in detected_objs]), frame)
             # if data_:
             #     print("dkld")