diff --git a/backend/models/yolo_manager.py b/backend/models/yolo_manager.py index f75d402..1d4512d 100644 --- a/backend/models/yolo_manager.py +++ b/backend/models/yolo_manager.py @@ -29,7 +29,7 @@ class YOLOManager: results = self.model.track(frame, persist=True, conf=conf, iou=iou, tracker="bytetrack.yaml", verbose=False) return results[0] - def detect(self, frame): + def detect(self, frame, conf=0.25): """Standard detection without tracking.""" - results = self.model.predict(frame, verbose=False) + results = self.model.predict(frame, conf=conf, verbose=False) return results[0]