导出部署模型fixed_input_shape后,预测时如何指定?
收藏
在测试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
收藏
请登录后评论
补充一下,是paddlex1.3的静态图模型
能想到的都试过了,比如[608,608],(608,608)……