From f48f320ab1516cf9293144e78ea19632cdb70880 Mon Sep 17 00:00:00 2001
From: baoshiwei <baoshiwei@shlanbao.cn>
Date: 星期日, 27 四月 2025 13:24:45 +0800
Subject: [PATCH] feat:优化安全检测与药材识别逻辑
---
openvino/safety_detect.py | 5 +++--
openvino/herb_ai.py | 3 +++
2 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/openvino/herb_ai.py b/openvino/herb_ai.py
index c916d9e..06734ff 100644
--- a/openvino/herb_ai.py
+++ b/openvino/herb_ai.py
@@ -230,6 +230,9 @@
res_ = "aidetect," + f"{det_res}"
logger.info("鍙戦�佸畨鍏ㄦ娴嬬粨鏋滐細"+str(res_))
l.send_msg(res_)
+ if 81 in class_ids:
+ logger.info("璇嗗埆鍒拌嵂鏉愶紝姝ｅ湪涓婃枡")
+ print("璇嗗埆鍒拌嵂鏉愶紝姝ｅ湪涓婃枡")
# 涓婃枡璇嗗埆
probabilities = load_identifier(frame)
diff --git a/openvino/safety_detect.py b/openvino/safety_detect.py
index f4d6e6d..c9a96e4 100644
--- a/openvino/safety_detect.py
+++ b/openvino/safety_detect.py
@@ -79,14 +79,15 @@
cv2.putText(img, label, (x - 10, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 2)
def draw_detections(self, frame, class_ids, scores, boxes):
+ det_img = frame.copy()
result_boxes = cv2.dnn.NMSBoxes(boxes, scores, 0.25, 0.45, 0.5)
for i in range(len(result_boxes)):
index = result_boxes[i]
box = boxes[index]
- self.draw_bounding_box(frame, class_ids[index], scores[index], round(box[0] * self.scale), round(box[1] * self.scale),
+ self.draw_bounding_box(det_img, class_ids[index], scores[index], round(box[0] * self.scale), round(box[1] * self.scale),
round((box[0] + box[2]) * self.scale), round((box[1] + box[3]) * self.scale))
- return frame
+ return det_img
--
Gitblit v1.9.3