changed yolo design

2026-05-30 19:29:43 +02:00
parent 66462c8f80
commit 4411fac977
4 changed files with 23 additions and 33 deletions
@@ -75,7 +75,7 @@ def use_webcam(grid, img_size):

        prediction = model(image)

-        bboxes, grid_ob, grid_noob = convert_prediction(prediction.squeeze(0), image.squeeze(0), threshold=0.9)
+        bboxes, grid_ob, grid_noob = convert_prediction(prediction.squeeze(0), image.squeeze(0), threshold=0.95)

        for bbox in grid_noob:
            xmin = int(bbox[0] * scale_w)