Toybrick

标题: 含上采样的onnx模型一律无法转换至rknn模型 [打印本页]

作者: magi803    时间: 2019-5-15 21:38
标题: 含上采样的onnx模型一律无法转换至rknn模型
急急急!只要神经网络里用了上采样,不论是插值的,还是pixshuffle,都会像如下一样报错,求大佬点播

作者: jefferyzhang    时间: 2019-5-17 15:42
目前的版本(1.0.0) onnx的上采样rknn无法支持, tensorflow是支持的。
作者: magi803    时间: 2019-5-17 17:22
jefferyzhang 发表于 2019-5-17 15:42
目前的版本(1.0.0) onnx的上采样rknn无法支持, tensorflow是支持的。

请问,pytorch代码先转为onnx,再把onnx转为tf来解决上采样问题可行吗?
作者: jefferyzhang    时间: 2019-5-17 17:47
magi803 发表于 2019-5-17 17:22
请问,pytorch代码先转为onnx,再把onnx转为tf来解决上采样问题可行吗?

理论可行,但是你每次转换后都要用对应环境跑下看看结果对不对,之前有盆友就是转换完onnx或者tf时候运行结果都不对了,再转rknn问题就更多了。。




欢迎光临 Toybrick (https://t.rock-chips.com/) Powered by Discuz! X3.3