onnx模型 获取中间节点数据

import onnx
import onnxruntime

model = onnx.load('simplenet.onnx')

#整个模型node添加到输出
#for node in model.graph.node:
#  for output in node.output:
#    model.graph.output.extend([onnx.ValueInfoProto(name=output)])

#指定node添加到输出

model.graph.output.extend([onnx.ValueInfoProto(name='288')])
sess = onnxruntime.InferenceSession(model.SerializeToString())

input_name = sess.get_inputs()[0].name

output_name = sess.get_outputs()[0].name
output1_name = sess.get_outputs()[1].name

#构造输入数据并推理

 outputs = sess.run([output1_name], {input_name: input_data})