rockchip-linux / rknn-toolkit2

BSD 3-Clause "New" or "Revised" License
828 stars 151 forks source link

ch_ppstructure_mobile_v2.0_SLANet表格识别模型转rknn模型报错 #302

Open cty-ai opened 3 months ago

cty-ai commented 3 months ago

模型

ch_ppstructure_mobile_v2.0_SLANet

转换过程

paddle->onnx->rknn

转换工具

paddle->onnx工具版本 : Paddle2ONNX Version:1.1.0
Paddle2ONNX now only support opset version in range of [7, 17]. onnx->rknn工具版本: rknn_toolkit2-1.5.3b21+11d8562a-cp38-cp38-linux_x86_64.whl rknn_toolkit2-1.6.0+81f21f4d-cp38-cp38-linux_x86_64.whl rknn_toolkit2-1.6.1b16+e3c0774a-cp38-cp38-linux_x86_64.whl rknn_toolkit2-1.6.2b3+eeda9fd5-cp38-cp38-linux_x86_64.whl 这几个toolkit都试过报一样的错误

问题描述和截图

paddle转onnx没报错且可以用onnxruntime正常推理,再转rknn报错如截图 image

zen-xingle commented 3 months ago

模型是动态shape的,请参考这个例子 https://github.com/airockchip/rknn-toolkit2/tree/master/rknn-toolkit2/examples/functions/dynamic_shape,配置需要的输入shape