Toybrick

load_onnx报错

yaowei

中级会员

积分
375
楼主
发表于 2019-5-15 09:36:46    查看: 7520|回复: 8 | [复制链接]    打印 | 显示全部楼层
rknn 1.0.0和0.9.9都试过。

模型文件:https://s3.amazonaws.com/onnx-model-zoo/arcface/resnet100.onnx


转换程序:
  1. from rknn.api import RKNN

  2. rknn = RKNN()

  3. rknn.config(channel_mean_value='127.5 127.5 127.5 128', reorder_channel='0 1 2')

  4. print('--> Loading model')
  5. rknn.load_onnx(model='./resnet100.onnx')
  6. print('done')

  7. print('--> Building model')
  8. rknn.build(do_quantization=False)
  9. print('done')

  10. rknn.export_rknn('./resnet100.rknn')

  11. rknn.release()
复制代码


报错:
  1. --> Loading model
  2. W Not match tensor MatMul_464:out0
  3. E Catch exception when loading onnx model: ./resnet100.onnx!
  4. T Traceback (most recent call last):
  5. T   File "rknn/api/rknn_base.py", line 344, in rknn.api.rknn_base.RKNNBase.load_onnx
  6. T   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 1037, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.match_paragraph_and_param
  7. T   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 950, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx._onnx_push_ready_tensor
  8. T TypeError: 'NoneType' object is not iterable
复制代码


回复

使用道具 举报

yaowei

中级会员

积分
375
沙发
 楼主| 发表于 2019-5-17 14:20:31 | 显示全部楼层
目前rknn版本1.0.0

  1. W verbose file path is invalid, debug info will not dump to file.
  2. --> Loading model
  3. I Current ONNX Model use ir_version 3 opset_version 8
  4. D import clients finished
  5. I build output layer attach_BatchNormalization_466:out0
  6. I Try match BatchNormalization_466:out0
  7. I Match r_bn_v6 [['BatchNormalization_466', 'Initializer_770', 'Initializer_771', 'Initializer_772', 'Initializer_773']] [['BatchNormalization', 'Constant_0', 'Constant_1', 'Constant_2', 'Constant_3']] to [['batchnormalize']]
  8. I Try match Add_465:out0
  9. I Match r_add [['Add_465']] [['Add']] to [['add']]
  10. I Try match MatMul_464:out0
  11. W Not match tensor MatMul_464:out0
  12. E Catch exception when loading onnx model: ./resnet100.onnx!
  13. T Traceback (most recent call last):
  14. T   File "rknn/api/rknn_base.py", line 344, in rknn.api.rknn_base.RKNNBase.load_onnx
  15. T   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 1037, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.match_paragraph_and_param
  16. T   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 950, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx._onnx_push_ready_tensor
  17. T TypeError: 'NoneType' object is not iterable
  18. done
  19. --> Building model
  20. E Model or data is None, please load model first.
  21. done
  22. E RKNN model data is None, please load model first!
复制代码
回复

使用道具 举报

yaowei

中级会员

积分
375
板凳
 楼主| 发表于 2019-5-17 14:23:40 | 显示全部楼层
感谢版版还记得这个帖子,现在onnx版本1.5.0,我降到过1.2.0,也是有问题
回复

使用道具 举报

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

本版积分规则

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


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