Toybrick

标题: onnx模型，build成功，init_runtime报错 [打印本页]

作者: MrMeetion 时间: 2022-12-23 17:00
标题: onnx模型，build成功，init_runtime报错
本帖最后由 MrMeetion 于 2022-12-23 17:02 编辑

1. RKNNToolKit版本：1.7.3
2. 运行模式：模拟器(target_platfrom为rv1126)
3. 报错信息：
超过发帖字节数限制，见附件图片1.png
4. onnx模型见附件
5. 报错信息分析：
报错信息提示在concat算子出现错误，通过rknn visualization工具，查看rknn模型，发现问题：
此处有图片，超过发帖字节数限制，见附件图片2.png
图片左侧是RKNN模型，1处的shape为(3, 1, 40, 512)， 2处的shape为(1, 1, 40, 512)。作为对比，图片右侧为onnx模型，3和4的shape都为(1, 512, 1, 40)。这里可以看出，1处的shape出现了错误。

请大神们看一看如何解决这个问题。

作者: MrMeetion 时间: 2022-12-30 10:51
没有大神指点迷津吗

欢迎光临 Toybrick (https://t.rock-chips.com/)