From 88fc0f9f9b7fd3eb81c958ca41ed822cf3657c47 Mon Sep 17 00:00:00 2001
From: baoshiwei <baoshiwei@shlanbao.cn>
Date: 星期二, 22 四月 2025 15:50:22 +0800
Subject: [PATCH] refactor: 重构中药识别项目 分为onnx版和openvino版
---
onnx/safety_detect.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/safety_detect.py b/onnx/safety_detect.py
similarity index 98%
rename from safety_detect.py
rename to onnx/safety_detect.py
index 61c5dd8..694a2de 100644
--- a/safety_detect.py
+++ b/onnx/safety_detect.py
@@ -18,7 +18,7 @@
return self.detect_objects(image)
def initialize_model(self, path):
- self.session = onnxruntime.InferenceSession(path, providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
+ self.session = onnxruntime.InferenceSession(path, providers=['CPUExecutionProvider'])
self.class_names = eval(self.session.get_modelmeta().custom_metadata_map['names'])
# Get model info
self.get_input_details()
--
Gitblit v1.9.3