Joydeep/Trrafic-Sign-and-Pothole_Detection
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test

Browse Source
This commit is contained in:
Joydeep Pandey 2026-02-09 18:18:15 +05:30
parent e0685b7e3b
commit 15d0350caa
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/models/yolo_manager.py
View File

@ -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]