UPDATE Vision 更新模型

2024-12-02 15:51:32 +08:00
parent 157e7edce5
commit e2821f7e48
9 changed files with 1228822 additions and 20 deletions
--- a/Vision/yolo/yolov8_pt_seg.py
+++ b/Vision/yolo/yolov8_pt_seg.py
@ -330,7 +330,7 @@ class yolov8_segment():
        preds = result[0]
        proto = result[1][-1]
        # NMS
-        det = non_max_suppression(preds, conf_thres=0.25, iou_thres=0.3, nc=len(self.model.CLASSES))[0]
+        det = non_max_suppression(preds, conf_thres=0.4, iou_thres=0.4, nc=len(self.model.CLASSES))[0]
        if det.shape[0] != 0:
            # bbox还原至原图尺寸
            det[:, :4] = scale_boxes(img.shape[2:], det[:, :4], ori_img.shape)