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