Firefly开源社区

标题: 使用 rknn-toolkit2，我的 onnx 模型转换 rknn 模型失败 [打印本页]
作者: hhdy    时间: 2023-4-20 11:45
标题: 使用 rknn-toolkit2，我的 onnx 模型转换 rknn 模型失败
本帖最后由 hhdy 于 2023-4-20 11:45 编辑

转换脚本（Demo中的 onnx 模型可以成功转换）：
  1. import os
  2. import urllib
  3. import traceback
  4. import time
  5. import sys
  6. import numpy as np
  7. from rknn.api import RKNN

  9. ONNX_MODEL = 'test4.onnx'
  10. RKNN_MODEL = 'test.rknn'

  12. if __name__ == '__main__':

  14.     rknn = RKNN(verbose=True)

  16.     # pre-process config
  17.     print('--> Config model')
  18.     rknn.config(target_platform='rk3588', optimization_level=0)
  19.     print('Config model done')

  21.     # Load ONNX model
  22.     print('--> Loading model')
  23.     ret = rknn.load_onnx( model=ONNX_MODEL )   #, inputs=['inputs', 'buffer'], input_size_list=[[514],[27724] ], outputs=['out', 'out_buffer'])  

  25.     if ret != 0:
  26.         print('Load model failed!')
  27.         exit(ret)
  28.     print('Loading model done')

  30.     # Build model
  31.     print('--> Building model')
  32.     ret = rknn.build(do_quantization=False)
  33.     if ret != 0:
  34.         print('Build model failed!')
  35.         exit(ret)
  36.     print('Building model done')

  38.     # Export RKNN model
  39.     print('--> Export rknn model')
  40.     ret = rknn.export_rknn(RKNN_MODEL)
  41.     if ret != 0:
  42.         print('Export rknn model failed!')
  43.         exit(ret)
  44.     print('Export rknn model done')

  46.     rknn.release()
复制代码


错误信息：
  1. W __init__: rknn-toolkit2 version: 1.4.0-22dcfef4
  2. --> Config model
  3. Config model done
  4. --> Loading model
  5. W load_onnx: The config.mean_values is None, zeros will be set for input 0!
  6. W load_onnx: The config.std_values is None, ones will be set for input 0!
  7. W load_onnx: The config.mean_values is None, zeros will be set for input 1!
  8. W load_onnx: The config.std_values is None, ones will be set for input 1!
  9. Loading model done
  10. --> Building model
  11. E build: Catch exception when building RKNN model!
  12. E build: Traceback (most recent call last):
  13. E build:   File "rknn/api/ir_graph.py", line 78, in rknn.api.ir_graph.IRGraph.infer_shapes
  14. E build:   File "/home/hhdy/.local/lib/python3.6/site-packages/onnx/shape_inference.py", line 37, in infer_shapes
  15. E build:     inferred_model_str = C.infer_shapes(model_str, check_type, strict_mode)
  16. E build: onnx.onnx_cpp2py_export.shape_inference.InferenceError: [ShapeInferenceError] (op_type:Reshape, node name: _0_expand1): [ShapeInferenceError] Inferred shape and existing shape differ in dimension 2: (1) vs (5)
  17. E build: During handling of the above exception, another exception occurred:
  18. E build: Traceback (most recent call last):
  19. E build:   File "rknn/api/rknn_base.py", line 1546, in rknn.api.rknn_base.RKNNBase.build
  20. E build:   File "rknn/api/graph_optimizer.py", line 1335, in rknn.api.graph_optimizer.GraphOptimizer.fuse_ops
  21. E build:   File "rknn/api/ir_graph.py", line 343, in rknn.api.ir_graph.IRGraph.rebuild
  22. E build:   File "rknn/api/ir_graph.py", line 185, in rknn.api.ir_graph.IRGraph._clean_model
  23. E build:   File "rknn/api/ir_graph.py", line 76, in rknn.api.ir_graph.IRGraph.infer_shapes
  24. E build:   File "rknn/api/ir_graph.py", line 84, in rknn.api.ir_graph.IRGraph.infer_shapes
  25. E build:   File "/home/hhdy/.local/lib/python3.6/site-packages/onnx/shape_inference.py", line 37, in infer_shapes
  26. E build:     inferred_model_str = C.infer_shapes(model_str, check_type, strict_mode)
  27. E build: onnx.onnx_cpp2py_export.shape_inference.InferenceError: [ShapeInferenceError] (op_type:Reshape, node name: _0_expand1): [ShapeInferenceError] Inferred shape and existing shape differ in dimension 2: (1) vs (5)
  28. Build model failed!
复制代码



tchip_askquestions

错误消息+日志.txt

32.23 KB, 下载次数: 1
作者: hhdy    时间: 2023-4-20 11:46
如果解决问题， 需要 我的模型， 也可以提供
作者: 板蓝根    时间: 2023-4-20 16:11
提供一下模型
作者: hhdy    时间: 2023-4-21 09:52
模型

gateconvnet.pt.sim.onnx

2.61 MB, 下载次数: 1
作者: hhdy    时间: 2023-4-21 11:15
板蓝根 发表于 2023-4-20 16:11
提供一下模型

大神  模型发出来了。 你看下
作者: hhdy    时间: 2023-4-21 19:15
本帖最后由 hhdy 于 2023-4-21 20:01 编辑

+++++
作者: icy    时间: 2023-7-24 15:41
问题解决了么？



欢迎光临 Firefly开源社区 (https://dev.t-firefly.com/) Powered by Discuz! X3.1