test
This commit is contained in:
parent
e0685b7e3b
commit
15d0350caa
@ -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]
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user