python 调用onnxruntime 实现单输入多输出

        做模型部署的时候需要将模型转换为onnx,转换好之后需要将图像传入验证对比原来的pth看输出参数是否一致。我的模型输出有三部分,直接贴出代码了,注释那部分可以实现指定输出部分。

import onnxruntime as ort
ort_session = ort.InferenceSession('./RetinaFace.onnx') #加载模型并且初始化
input_name = ort_session.get_inputs()[0].name 
# outputs_1 = ort_session.get_outputs()[0].name
# outputs_2 = ort_session.get_outputs()[0].name
# out = ort_session.run([outputs_1], input_feed={input_name: image1})
#run第一个关键字设置None,image1为模型输入,一般需要转为numpy格式,如:image1 = image.numpy()
outs = ort_session.run(None, input_feed={input_name: image1}) 
print('out_0:',out[0])
print('out_1:',out[1])
print('out_2:',out[2])

贴上两个可以深入学习onnx的连接:

Python onnxruntime.InferenceSession方法代码示例 - 纯净天空

onnxruntime的C++ api如何实现session的多输入与多输出? - 知乎

但是问题又来了,多模态的视觉任务中,如何使用onnxruntime实现多输入多输出呢?

你可能感兴趣的:(模型部署,深度学习,神经网络,python)