set conf = 0.1 for the yolov10s detection model

OneMagicKey · Jun 16, 2024 · 109237c · 109237c
1 parent 7ce9737
commit 109237c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/bot.py b/src/bot.py
@@ -395,8 +395,8 @@ def main():
 
 
 model_list = [
-    ModelInfo("detection", "yolov10s", (640, 640), 10),
-    ModelInfo("segmentation", "yolov8s-seg", (640, 640), 8),
+    ModelInfo("detection", "yolov10s", (640, 640), conf=0.1, version=10),
+    ModelInfo("segmentation", "yolov8s-seg", (640, 640), version=8),
 ]
 
 if __name__ == "__main__":