Toybrick

onnx转rknn报错

liuwenzhen

注册会员

积分
126
楼主
发表于 2020-5-29 10:50:23    查看: 11723|回复: 9 | [复制链接]    打印 | 只看该作者
环境:Debian10 rknn-toolkit1.3.2
问题:onnx转rknn过程中,mobilenetv1可以转换成功,mobilenetv2 v3转换报错KeyError: 'data_bias',这个报错的意思是模型字典里面没有字典‘data_bais’吗?请问为什么v1可以成功v2 v3确不可以?
回复

使用道具 举报

Aiden

注册会员

积分
88
沙发
发表于 2020-5-30 01:24:25 | 只看该作者
我也遇到过,是 pytorch 的模型在 optimizaiton_level=0 的时候遇到的。感觉好艰难啊。
回复

使用道具 举报

liuwenzhen

注册会员

积分
126
板凳
 楼主| 发表于 2020-5-30 12:14:32 | 只看该作者
Aiden 发表于 2020-5-30 01:24
我也遇到过,是 pytorch 的模型在 optimizaiton_level=0 的时候遇到的。感觉好艰难啊。 ...

那这么说,就是不优化不行了,感觉还有好多都不支持,而且找不到原因
回复

使用道具 举报

jefferyzhang

版主

积分
13580
地板
发表于 2020-5-30 17:50:20 | 只看该作者
liuwenzhen 发表于 2020-5-30 12:14
那这么说,就是不优化不行了,感觉还有好多都不支持,而且找不到原因

pytorch刚开始支持,目前都是在收集常用的op,然后不断的更新加进去。。。
tf和caffe的支持会好一点。

那个optimazition_level不是优化不优化的设置,不设置该标志时候如果推理结果是正确的,就别去改他。
有些时候发现op被错误的融合导致输出结果不正确时候,就必须把这个设成0来尝试是否能解决问题。
回复

使用道具 举报

kyo055

注册会员

积分
87
5#
发表于 2020-6-2 17:14:34 | 只看该作者
jefferyzhang 发表于 2020-5-30 17:50
pytorch刚开始支持,目前都是在收集常用的op,然后不断的更新加进去。。。
tf和caffe的支持会好一点。

请问rknn支持shufflechannel层吗? 如何查看rknn支持的ops?
回复

使用道具 举报

jefferyzhang

版主

积分
13580
6#
发表于 2020-6-2 17:51:24 | 只看该作者
kyo055 发表于 2020-6-2 17:14
请问rknn支持shufflechannel层吗? 如何查看rknn支持的ops?

看不到,你就当都支持,verbose打开,有不支持的会提示你。
然后你可以想办法绕过去,或者报给我们这个op,我们都会安排去支持
回复

使用道具 举报

kyo055

注册会员

积分
87
7#
发表于 2020-6-2 18:30:51 | 只看该作者
jefferyzhang 发表于 2020-6-2 17:51
看不到,你就当都支持,verbose打开,有不支持的会提示你。
然后你可以想办法绕过去,或者报给我们这个op ...

OK 我尝试一下 这边是ShuffleNet 所以不清楚能不能支持
回复

使用道具 举报

liuwenzhen

注册会员

积分
126
8#
 楼主| 发表于 2020-6-4 11:21:27 | 只看该作者
kyo055 发表于 2020-6-2 18:30
OK 我尝试一下 这边是ShuffleNet 所以不清楚能不能支持

我尝试过,不能支持shufflenet,因为不支持超过四维的处理,shuffle操作是五维的
回复

使用道具 举报

kyo055

注册会员

积分
87
9#
发表于 2020-6-5 10:39:28 | 只看该作者
liuwenzhen 发表于 2020-6-4 11:21
我尝试过,不能支持shufflenet,因为不支持超过四维的处理,shuffle操作是五维的 ...

我的ShuffleNet Caffe居然转成了,但现在另一个问题很尴尬,就是运行时报错
回复

使用道具 举报

kyo055

注册会员

积分
87
10#
发表于 2020-6-5 10:40:41 | 只看该作者
jefferyzhang 发表于 2020-6-2 17:51
看不到,你就当都支持,verbose打开,有不支持的会提示你。
然后你可以想办法绕过去,或者报给我们这个op ...

--> Running model
ASSERT in NeuralNet.cpp.decompressKernel(1889): readBias == 0
terminate called after throwing an instance of 'bool'

请问rknn.inference(inputs=[img_bgr]) 这个地方前传报错了,是怎么回事
回复

使用道具 举报

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

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


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