UPDATE Vision 更新openvino模型

2024-12-02 17:20:40 +08:00
parent e2821f7e48
commit dc640f461d
8 changed files with 172 additions and 140 deletions
--- a/Vision/detect_person.py
+++ b/Vision/detect_person.py
@ -49,6 +49,7 @@ class DetectionPerson:
        if ret:
            # img_src = cv2.imread(img_path)
            img = self.precess_image(img_src, self.imgsz, self.half, self.cuda)
+            img = img[250:1000, 380:2000]
            preds = self.model(img)
            det = ops.non_max_suppression(preds, self.conf, self.iou, classes=None, agnostic=False, max_det=300,
                                          nc=len(self.names))