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实现多输入多输出呢?

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值