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( [