onnx 模型操作相关

1. 检查模型

# pip install onnx
import onnx
# 加载模型
model = onnx.load('model.onnx')
# 检查模型格式是否完整及正确
onnx.checker.check_model(model)
print(model.graph.output)

2. 推理

import onnxruntime as ort

sess = ort.InferenceSession('model.onnx')
inputs = {
    # None 处一般为ndarray 等
    sess.get_inputs()[0].name: None
}
outputs = sess.run(None, inputs)

3. 例子

RMBG1.4