含上采样的onnx模型一律无法转换至rknn模型

[复制链接] · 发表于 2019-5-15 21:38:03

急急急！只要神经网络里用了上采样，不论是插值的，还是pixshuffle，都会像如下一样报错，求大佬点播

只看该作者 · 发表于 2019-5-17 15:42:46

目前的版本（1.0.0） onnx的上采样rknn无法支持， tensorflow是支持的。

只看该作者 · 发表于 2019-5-17 17:22:34

jefferyzhang 发表于 2019-5-17 15:42
目前的版本（1.0.0） onnx的上采样rknn无法支持， tensorflow是支持的。

请问，pytorch代码先转为onnx，再把onnx转为tf来解决上采样问题可行吗？

只看该作者 · 发表于 2019-5-17 17:47:41

magi803 发表于 2019-5-17 17:22
请问，pytorch代码先转为onnx，再把onnx转为tf来解决上采样问题可行吗？

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