python 运行 onnx模型 进行推理的代码
时间: 2023-05-20 15:01:10 浏览: 660
onnx-ml-demo:使用ONNX进行模型推理的演示
可以使用 onnxruntime 库来运行 onnx 模型进行推理,以下是一个简单的 Python 代码示例:
```python
import onnxruntime as ort
import numpy as np
# 加载模型
sess = ort.InferenceSession('model.onnx')
# 准备输入数据
input_data = np.random.randn(1, 3, 224, 224).astype(np.float32)
# 进行推理
output = sess.run(None, {'input': input_data})
# 输出结果
print(output)
```
其中,`model.onnx` 是你的 onnx 模型文件名,`input` 是模型的输入名称,`input_data` 是输入数据,`output` 是模型的输出结果。
阅读全文