Toybrick

由onnx格式转为rknn格式的过程中一直报错

jacksun

新手上路

积分
9
发表于 2020-11-16 12:58:00    查看: 590|回复: 0 | [复制链接]    打印 | 显示全部楼层
目前我们利用paddlex算法训练的模型,算法训练好后,我们利用paddle2onnx包进行了模型转换,转为onnx格式。目前在由onnx格式转为rknn格式的过程中一直报错,报错信息如下:

T Traceback (most recent call last):
T File "rknn/api/rknn_base.py", line 344, in rknn.api.rknn_base.RKNNBase.load_onnx
T File "rknn/base/RKNNlib/converter/convert_onnx.py", line 523, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.__init__
T File "rknn/base/RKNNlib/converter/convert_onnx.py", line 529, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.__init__
T File "rknn/base/RKNNlib/converter/onnx_util.py", line 158, in rknn.base.RKNNlib.converter.onnx_util.ONNXProto_Util.polishing_model
T File "/Users/song.li/anaconda3/envs/pythonLearn/lib/python3.6/site-packages/onnx/utils.py", line 18, in polish_model
T onnx.checker.check_model(model)
T File "/Users/song.li/anaconda3/envs/pythonLearn/lib/python3.6/site-packages/onnx/checker.py", line 86, in check_model
T C.check_model(model.SerializeToString())
T onnx.onnx_cpp2py_export.checker.ValidationError: Your model ir_version is higher than the checker's.
done
--> Building model
Traceback (most recent call last):
File "rknn_build.py", line 35, in <module>
rknn.build(do_quantization=False)
File "/Users/song.li/anaconda3/envs/pythonLearn/lib/python3.6/site-packages/rknn/api/rknn.py", line 163, in build
inputs = self.rknn_base.net.get_input_layers()
AttributeError: 'NoneType' object has no attribute 'get_input_layers'

我们目前使用的版本信息如下:

tensorflow 1.14.0,
rknn_toolkit-1.2.0 ( rknn_toolkit-1.2.0-cp36-cp36m-macosx_10_9_x86_64.whl )
paddle2onnx 0. 4

    请教相应的解决方法。谢谢!

回复

使用道具 举报

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

本版积分规则

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


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