首页 PaddleX 帖子详情
导出部署模型fixed_input_shape后,预测时如何指定?
收藏
快速回复
PaddleX 问答入门 482 2
导出部署模型fixed_input_shape后,预测时如何指定?
收藏
快速回复
PaddleX 问答入门 482 2

在测试PaddleX用onnxruntime部署,但是模型的输入大小需要固定才能使用Paddle2ONNX成功转换,故而导出时通过指定--fixed_input_shape固定大小。

但是现在问题来了,onnxruntime的session要run的时候,im_size要怎么指定啊?

for i in range(len(sess.get_inputs())):
    print(sess.get_inputs()[i].name,sess.get_inputs()[i].shape)
image [None, 3, 640, 640]
im_size [None, 2]

就是这里……

sess.run(None,{'image':image,'im_size':???})
0
收藏
回复
全部评论(2)
时间顺序
深渊上的坑
#2 回复于2021-09

补充一下,是paddlex1.3的静态图模型

0
回复
深渊上的坑
#3 回复于2021-09

能想到的都试过了,比如[608,608],(608,608)……

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户