onnx 测试_Onnx模型推理（Python）,onnx,python

# -*-coding: utf-8 -*-

import os, sys

sys.path.append(os.getcwd())

import onnxruntime

import onnx

class ONNXModel():

def __init__(self, onnx_path):

"""

:param onnx_path:

"""

self.onnx_session = onnxruntime.InferenceSession(onnx_path)

self.input_name = self.get_input_name(self.onnx_session)

self.output_name = self.get_output_name(self.onnx_session)

print("input_name:{}".format(self.input_name))

print("output_name:{}".format(self.output_name))

def get_output_name(self, onnx_session):

"""

output_name = onnx_session.get_outputs()[0].name

:param onnx_session:

:return:

"""

output_name = []

for node in onnx_session.get_outputs():

output_name.append(node.name)

retu