onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Invalid Feed Input Name:input1
时间: 2023-10-26 19:14:52 浏览: 275
c++ std::invalid_argument应用
5星 · 资源好评率100%
这个错误提示表明输入数据的名称不正确。在使用 ONNX Runtime 运行模型时,需要指定输入数据的名称和对应的数据。请检查输入数据的名称是否正确,并确保与模型中定义的输入数据名称相同。你可以使用下面的代码来检查模型的输入数据名称:
```
import onnxruntime as ort
model_path = "path/to/your/model.onnx"
sess = ort.InferenceSession(model_path)
input_name = sess.get_inputs()[0].name
print("Model input name:", input_name)
```
如果你在指定输入数据时使用了错误的名称,可以使用上面的代码来获取正确的输入数据名称。
阅读全文