Toybrick

标题: onnx转rknn成功，但是init_runtime失败 [打印本页]

作者: ruiyanlpf 时间: 2019-10-29 21:04
标题: onnx转rknn成功，但是init_runtime失败
设备信息如图所示，我的onnx模型是基于pytorch模型转过来的，pytorch版本1.2.0。然后使用load_onnx转换模型成功，但是init_runtime模型时报下面的错误。代码如图所示。

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

作者: ruiyanlpf 时间: 2019-10-31 12:33

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

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

作者: rlgl 时间: 2020-4-1 13:50

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

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

作者: 18022443868 时间: 2020-5-6 17:40
同求，我也想要对应的转化py文件

作者: Yaphet 时间: 2021-9-8 14:17
我也遇到了该问题，能否提供代码分享下，十分感谢

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