onnx转rknn成功，但是init_runtime失败

[复制链接] · 发表于 2019-10-29 21:04:35

设备信息如图所示，我的onnx模型是基于pytorch模型转过来的，pytorch版本1.2.0。然后使用load_onnx转换模型成功，但是init_runtime模型时报下面的错误。代码如图所示。

只看该作者 · 发表于 2019-10-29 22:26:25

转换rknn过程如果没有报错，一般来说就是pytorch转onnx时候出了问题。
建议你onnx的模型找个方法读出来推理下看看有没啥错。
也可以试试在不同环境下init_runtime会不会有不同结果，例如x86上仿真和直接在3399pro上init_runtime。

只看该作者 · 发表于 2019-10-31 12:33:52

jefferyzhang 发表于 2019-10-29 22:26
转换rknn过程如果没有报错，一般来说就是pytorch转onnx时候出了问题。
建议你onnx的模型找个方法读出来推理 ...

谢谢，已经解决，发现是模型转onnx存在的问题。

只看该作者 · 发表于 2020-4-1 13:50:04

ruiyanlpf 发表于 2019-10-31 12:33
谢谢，已经解决，发现是模型转onnx存在的问题。

请问能告诉下您pytorch模型转onnx的代码吗，我现在也在做这个转换，转换出来的onnx无法转换为rknn模型，感觉应该是我的转换代码不对（pytorch转onnx）

只看该作者 · 发表于 2020-5-6 17:40:54

同求，我也想要对应的转化py文件

只看该作者 · 发表于 2021-9-8 14:17:02

我也遇到了该问题，能否提供代码分享下，十分感谢