sergey21000 commited on
Commit
cd57024
·
verified ·
1 Parent(s): ed01d72

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -55,8 +55,8 @@ def detect_video(video_path_or_url: str, model: YOLO, conf: float, iou: float) -
55
 
56
  generator = model.predict(
57
  source=video_path,
58
- conf=0.5,
59
- iou=0.5,
60
  save=True,
61
  save_txt=True,
62
  save_conf=True,
 
55
 
56
  generator = model.predict(
57
  source=video_path,
58
+ conf=conf,
59
+ iou=iou,
60
  save=True,
61
  save_txt=True,
62
  save_conf=True,