UPDATE Vision 新增Nan值判断

2024-09-24 09:34:59 +08:00
parent 41cf46e71d
commit 5295630001
6 changed files with 165 additions and 71 deletions
--- a/Vision/yolo/yolov8_openvino.py
+++ b/Vision/yolo/yolov8_openvino.py
@ -42,6 +42,9 @@ class yolov8_segment_openvino:

        self.boxes, self.scores, self.class_ids, mask_pred = self.process_box_output(outputs[0])
        self.mask_maps = self.process_mask_output(mask_pred, outputs[1])
+        if type(self.boxes) == list:
+            if len(self.boxes)==0:
+                return 0, None, None, None, None
        self.boxes = self.boxes.tolist()
        self.scores = self.scores.tolist()
        for t in range(len(self.scores)):