zjt/SAMapi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix onnxruntime load error for newest onnxruntime

Browse Source
This commit is contained in:
triple-Mu 2023-04-06 09:20:31 +08:00 committed by ericmintun
parent dec4c12940
commit 6abd03257a
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/export_onnx_model.py
View File

@ -160,7 +160,9 @@ def run_export(
if onnxruntime_exists: if onnxruntime_exists:
ort_inputs = {k: to_numpy(v) for k, v in dummy_inputs.items()} ort_inputs = {k: to_numpy(v) for k, v in dummy_inputs.items()}
ort_session = onnxruntime.InferenceSession(output) # set cpu provider default
providers = ['CPUExecutionProvider']
ort_session = onnxruntime.InferenceSession(output, providers=providers)
_ = ort_session.run(None, ort_inputs) _ = ort_session.run(None, ort_inputs)
print("Model has successfully been run with ONNXRuntime.") print("Model has successfully been run with ONNXRuntime.")