From 67751fe874caa59e967b427ced3294751249c086 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=99=BD=E5=8F=B6=20=E8=97=A4=E5=8E=9F?= <1751842477@qq.com> Date: Tue, 19 Dec 2023 23:13:50 +0800 Subject: [PATCH] fix --- onnx_infer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onnx_infer.py b/onnx_infer.py index b58125d..5f414ac 100644 --- a/onnx_infer.py +++ b/onnx_infer.py @@ -1,4 +1,4 @@ -from onnx_modules.V220_OnnxInference import OnnxInferenceSession +锘縡rom onnx_modules.V220_OnnxInference import OnnxInferenceSession import numpy as np Session = OnnxInferenceSession( { @@ -12,7 +12,7 @@ Session = OnnxInferenceSession( Providers = ["CPUExecutionProvider"] ) -#这里的输入和原版是一样的,只需要在原版预处理结果出来之后加上.numpy()即可 +#杩欓噷鐨勮緭鍏ュ拰鍘熺増鏄竴鏍风殑锛屽彧闇瑕佸湪鍘熺増棰勫鐞嗙粨鏋滃嚭鏉ヤ箣鍚庡姞涓.numpy()鍗冲彲 x = np.expand_dims( np.array( [