Onnx 模型转换失败 rknn 1.6.0

[复制链接] · 发表于 2021-3-9 11:20:32

本帖最后由 jww 于 2021-3-9 11:23 编辑

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

toybrick@debian10:~/solo$ python3 test.py 

--> Config model

done

--> Loading model



W Not match tensor Slice_917:out0 

E Try match Slice_917:out0 failed, catch exception!

W ----------------Warning(1)----------------

E Catch exception when loading onnx model: piggie.shapes.onnx!

E Traceback (most recent call last):

E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 826, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.match_paragraph_and_param

E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 728, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx._onnx_push_ready_tensor

E TypeError: 'NoneType' object is not iterable

E During handling of the above exception, another exception occurred:

E Traceback (most recent call last):

E   File "rknn/api/rknn_base.py", line 264, in rknn.api.rknn_base.RKNNBase.load_onnx

E   File "rknn/base/RKNNlib/RK_nn.py", line 135, in rknn.base.RKNNlib.RK_nn.RKnn.load_onnx

E   File "rknn/base/RKNNlib/app/importer/import_onnx.py", line 121, in rknn.base.RKNNlib.app.importer.import_onnx.Importonnx.run

E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 832, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.match_paragraph_and_param

E   File "rknn/api/rknn_log.py", line 312, in rknn.api.rknn_log.RKNNLog.e

E ValueError: Try match Slice_917:out0 failed, catch exception!

Load resnet50v2 failed!
复制代码

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

只看该作者 · 发表于 2021-3-9 15:41:10

Try match Slice_917

ut0，LOG显示是这个OP不支持。
可以尝试自己修改下模型。
建议到置顶帖rknn 官方qq群里讨论下。

只看该作者 · 发表于 2021-3-10 14:06:56

确实是 Slice 层不支持，但 Slice 在 1.6.0 支持的列表里面，所以很奇怪了。
不过我把 Slice 层砍掉之后问题解决了，因为我的网络刚好这个 Slice 可以调整省掉，所以网络可以转换了。

Onnx 模型转换失败 rknn 1.6.0

本帖子中包含更多资源