This commit is contained in:
Joydeep Pandey 2026-02-09 18:18:15 +05:30
parent e0685b7e3b
commit 15d0350caa

View File

@ -29,7 +29,7 @@ class YOLOManager:
results = self.model.track(frame, persist=True, conf=conf, iou=iou, tracker="bytetrack.yaml", verbose=False) results = self.model.track(frame, persist=True, conf=conf, iou=iou, tracker="bytetrack.yaml", verbose=False)
return results[0] return results[0]
def detect(self, frame): def detect(self, frame, conf=0.25):
"""Standard detection without tracking.""" """Standard detection without tracking."""
results = self.model.predict(frame, verbose=False) results = self.model.predict(frame, conf=conf, verbose=False)
return results[0] return results[0]