Toybrick

ONNX转RKNN出现Slice不支持的问题

Pony

注册会员

积分
138
发表于 2020-12-23 17:19:04    查看: 11226|回复: 4 | [复制链接]    打印 | 显示全部楼层

运行环境:台式机上的CentOS
RKNN-TOOLKIT:1.4.0
目的:将NanoDet模型转成RKNN, 已经从PyTorch转成了ONNX
指令和输出:
(rknn) [jtma3@localhost nanodet]$ python test.py
--> config model
done
--> Loading model
/home/jtma3/Downloads/onnx-tensorflow-1.2.1/onnx_tf/common/__init__.py:87: UserWarning: FrontendHandler.get_outputs_names is deprecated. It will be removed in future release.. Use node.outputs instead.
  warnings.warn(message)
E Catch exception when loading onnx model: nanodet_1_3_320_320.onnx!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 556, in rknn.api.rknn_base.RKNNBase.load_onnx
E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 497, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.__init__
E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 500, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.__init__
E   File "/home/jtma3/anaconda3/envs/rknn/lib/python3.6/site-packages/onnx/checker.py", line 86, in check_model
E     C.check_model(model.SerializeToString())
E onnx.onnx_cpp2py_export.checker.ValidationError: Node () has input size 4 not in range [min=1, max=1].
E ==> Context: Bad node spec: input: "567" input: "569" input: "570" input: "568" output: "571" op_type: "Slice"
Load nanodet failed!









所以,RKNN目前是不支持slice操作码?转过其他模型也报过类似的错误,卡在了slice这里
回复

使用道具 举报

zhw

注册会员

积分
107
发表于 2020-12-29 14:50:20 | 显示全部楼层
你好,问题解决了吗
回复

使用道具 举报

Pony

注册会员

积分
138
 楼主| 发表于 2021-1-7 15:49:48 | 显示全部楼层
zhw 发表于 2020-12-29 14:50
你好,问题解决了吗

解决了,RKNN对于ONNX的模型实际上是支持的,但是有的模型里面会出现slice的问题,这个时候可以看看能不能在slice层后面加一个没有用的层,然后就不报错了。
回复

使用道具 举报

matrix

注册会员

积分
106
发表于 2021-1-22 17:54:15 | 显示全部楼层
请问pony如何加没用的层?能给个例子吗? 我也遇到同样的问题。rknn 1.6
pytorch 1.6
回复

使用道具 举报

另一个你呀

新手上路

积分
42
发表于 2021-11-8 20:13:41 | 显示全部楼层
Pony 发表于 2021-1-7 15:49
解决了,RKNN对于ONNX的模型实际上是支持的,但是有的模型里面会出现slice的问题,这个时候可以看看能不 ...

请问怎么加没用的层呢?
回复

使用道具 举报

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

本版积分规则

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


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