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/identifier.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/identifier.py b/onnx/identifier.py
similarity index 97%
rename from identifier.py
rename to onnx/identifier.py
index 293bffe..b86d734 100644
--- a/identifier.py
+++ b/onnx/identifier.py
@@ -14,7 +14,7 @@
return self.idengify(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