This commit is contained in:
白叶 藤原
2023-12-19 23:13:50 +08:00
parent d7855d74b0
commit 67751fe874

View File

@@ -1,4 +1,4 @@
from onnx_modules.V220_OnnxInference import OnnxInferenceSession
from 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(
[