Toybrick

Onnx 模型转换失败 rknn 1.6.0

jww

新手上路

积分
39
楼主
发表于 2021-3-9 11:20:32    查看: 14990|回复: 2 | [复制链接]    打印 | 只看该作者
本帖最后由 jww 于 2021-3-9 11:23 编辑

我们首先运行了 resnet50.onnx 的 demo,没问题之后尝试转换我们自己的模型,碰到如下日志的报错

  1. toybrick@debian10:~/solo$ python3 test.py
  2. --> Config model
  3. done
  4. --> Loading model

  5. W Not match tensor Slice_917:out0
  6. E Try match Slice_917:out0 failed, catch exception!
  7. W ----------------Warning(1)----------------
  8. E Catch exception when loading onnx model: piggie.shapes.onnx!
  9. E Traceback (most recent call last):
  10. E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 826, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.match_paragraph_and_param
  11. E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 728, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx._onnx_push_ready_tensor
  12. E TypeError: 'NoneType' object is not iterable
  13. E During handling of the above exception, another exception occurred:
  14. E Traceback (most recent call last):
  15. E   File "rknn/api/rknn_base.py", line 264, in rknn.api.rknn_base.RKNNBase.load_onnx
  16. E   File "rknn/base/RKNNlib/RK_nn.py", line 135, in rknn.base.RKNNlib.RK_nn.RKnn.load_onnx
  17. E   File "rknn/base/RKNNlib/app/importer/import_onnx.py", line 121, in rknn.base.RKNNlib.app.importer.import_onnx.Importonnx.run
  18. E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 832, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.match_paragraph_and_param
  19. E   File "rknn/api/rknn_log.py", line 312, in rknn.api.rknn_log.RKNNLog.e
  20. E ValueError: Try match Slice_917:out0 failed, catch exception!
  21. Load resnet50v2 failed!
复制代码

这里面提到的 Slice_917 out0 并没有什么特殊的地方,网络结构如下图:




本帖子中包含更多资源

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

x
回复

使用道具 举报

jefferyzhang

版主

积分
13580
沙发
发表于 2021-3-9 15:41:10 | 只看该作者
Try match Slice_917ut0,LOG显示是这个OP不支持。
可以尝试自己修改下模型。
建议到置顶帖rknn 官方qq群里讨论下。
回复

使用道具 举报

jww

新手上路

积分
39
板凳
 楼主| 发表于 2021-3-10 14:06:56 | 只看该作者
确实是 Slice 层不支持,但 Slice 在 1.6.0 支持的列表里面,所以很奇怪了。
不过我把 Slice 层砍掉之后问题解决了,因为我的网络刚好这个 Slice 可以调整省掉,所以网络可以转换了。
回复

使用道具 举报

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

本版积分规则

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


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