tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
白叶 藤原
2023-12-19 23:13:50 +08:00
parent d7855d74b0
commit 67751fe874
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
onnx_infer.py
View File

@@ -1,4 +1,4 @@
from onnx_modules.V220_OnnxInference import OnnxInferenceSession from onnx_modules.V220_OnnxInference import OnnxInferenceSession
import numpy as np import numpy as np
Session = OnnxInferenceSession( Session = OnnxInferenceSession(
{ {
@@ -12,7 +12,7 @@ Session = OnnxInferenceSession(
Providers = ["CPUExecutionProvider"] Providers = ["CPUExecutionProvider"]
) )
#这里的输入和原版是一样的，只需要在原版预处理结果出来之后加上.numpy()即可 #这里的输入和原版是一样的，只需要在原版预处理结果出来之后加上.numpy()即可
x = np.expand_dims( x = np.expand_dims(
np.array( np.array(
[ [