Toybrick

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

magi803

新手上路

积分
32
发表于 2019-5-15 21:38:03    查看: 139|回复: 3 | [复制链接]    打印 | 显示全部楼层
急急急!只要神经网络里用了上采样,不论是插值的,还是pixshuffle,都会像如下一样报错,求大佬点播

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

jefferyzhang

超级版主

积分
758
发表于 2019-5-17 15:42:46 | 显示全部楼层
目前的版本(1.0.0) onnx的上采样rknn无法支持, tensorflow是支持的。
回复

使用道具 举报

magi803

新手上路

积分
32
 楼主| 发表于 2019-5-17 17:22:34 | 显示全部楼层
jefferyzhang 发表于 2019-5-17 15:42
目前的版本(1.0.0) onnx的上采样rknn无法支持, tensorflow是支持的。

请问,pytorch代码先转为onnx,再把onnx转为tf来解决上采样问题可行吗?
回复

使用道具 举报

jefferyzhang

超级版主

积分
758
发表于 2019-5-17 17:47:41 | 显示全部楼层
magi803 发表于 2019-5-17 17:22
请问,pytorch代码先转为onnx,再把onnx转为tf来解决上采样问题可行吗?

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

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 wiki教程 关于Toybrick


快速回复 返回顶部 返回列表