如何在Python上运行ONNX模型文件

0 人关注

我有一个ONNX模型文件，其中包含文本分类器。我也有数据，我的目的是在新的数据上测试该模型。但是我无法运行这些代码。首先，我尝试了下面的代码，但什么也没有发生

model_name = 'text_model.onnx'
onnx_model = onnx.load(model_name)
onnx.checker.check_model(onnx_model)
Then I try this
ort.InferenceSession(onnx_model, None)
但我得到一个错误。
TypeError: Unable to load from type '<class 'onnx.onnx_ml_pb2.ModelProto'>'
我应该怎么做？


         
          
           python


         
          
           machine-learning


         
          
           onnx


          
           
            已采纳


          
           
            
             对我有用的是以下代码
            
            from onnxruntime import InferenceSession
filename = "./svm_iris.onnx"
def execute_onnx_model_from_file(filename: str) -> None:
    sess = InferenceSession(filename)
    x_test, y_test = json_to_ndarray()
    sess.run(None, {"X": x_test.astype(np.float32)})[0]
In the 文件它规定，你可以将存储的ONNX-模型的文件名交给InferenceSession。另一个更接近你的代码的解决方案是将onnx-model序列化。
from onnxruntime import InferenceSession
filename = "./svm_iris.onnx"
def execute_onnx_model_from_file(filename: str) -> None: