Toybrick

标题: Onnx 模型转换失败 rknn 1.6.0 [打印本页]

作者: jww    时间: 2021-3-9 11:20
标题: Onnx 模型转换失败 rknn 1.6.0
本帖最后由 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 并没有什么特殊的地方,网络结构如下图:
[attach]1719[/attach]




作者: jefferyzhang    时间: 2021-3-9 15:41
Try match Slice_917ut0,LOG显示是这个OP不支持。
可以尝试自己修改下模型。
建议到置顶帖rknn 官方qq群里讨论下。
作者: jww    时间: 2021-3-10 14:06
确实是 Slice 层不支持,但 Slice 在 1.6.0 支持的列表里面,所以很奇怪了。
不过我把 Slice 层砍掉之后问题解决了,因为我的网络刚好这个 Slice 可以调整省掉,所以网络可以转换了。




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