Toybrick

标题: onnx转换rknn,所有的坑 [打印本页]

作者: kepurSong    时间: 2019-9-15 16:14
标题: onnx转换rknn,所有的坑
本帖最后由 kepurSong 于 2019-9-15 16:15 编辑

1,Not match tensor Upsample_36ut0‘NoneType’ object is not iterable!



作者: jefferyzhang    时间: 2019-9-15 20:19
看log显示是Upsample OP不支持
作者: kepurSong    时间: 2019-9-16 10:03
jefferyzhang 发表于 2019-9-15 20:19
看log显示是Upsample OP不支持

对的,这个不支持一般怎么解决?

作者: jefferyzhang    时间: 2019-9-17 09:02
kepurSong 发表于 2019-9-16 10:03
对的,这个不支持一般怎么解决?

一般:
1. 改模型换OP
2. 转rknn时候output移到支持的点,然后不支持的用cpu写
3. 自定义OP支持
作者: kepurSong    时间: 2019-9-17 12:08
标题: pb模型转换时,input和output如何写?
本帖最后由 kepurSong 于 2019-9-17 12:13 编辑
jefferyzhang 发表于 2019-9-17 09:02
一般:
1. 改模型换OP
2. 转rknn时候output移到支持的点,然后不支持的用cpu写

pb模型转换rknn:
我生成了一个pb模型,用netron查看了结构,但是不知道input_node和output_node怎么写?

作者: jefferyzhang    时间: 2019-9-17 15:11
点node,netron右边有写name那个,就是节点名字
作者: kepurSong    时间: 2019-9-17 16:13
jefferyzhang 发表于 2019-9-17 15:11
点node,netron右边有写name那个,就是节点名字

感谢高手指点,我bp转rknn终于成功了!
我要试试效果了!




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