Firefly开源社区

[人工智能] rknn-toolkit2模型转化失败

46

积分

0

威望

0

贡献

技术小白

积分
46
发表于 2021-8-24 13:15:48     
  1. import numpy as np
  2. import re
  3. import math
  4. import random
  5. import cv2
  6. from rknn.api import RKNN
  7. rknn = RKNN()
  8. rknn.config(mean_values=[128,128,128,128], std_values=[0,0,0,0])
  9. rknn.load_tensorflow(tf_pb='./foods26_frozen.pb',
  10.                      inputs=['input'],
  11.                      outputs=['InceptionV3/Predictions/Reshape_1'],
  12.                      input_size_list=[[299, 299, 3]])
  13. rknn.build(do_quantization=True, dataset='./dataset.txt')
  14. rknn.export_rknn('./test.rknn')
复制代码
上面是转化代码,直接报错
结果build时报错:
E build: Catch exception when building RKNN model!
E build: Traceback (most recent call last):
E build: File "rknn/api/rknn_base.py", line 2254, in rknn.api.rknn_base.RKNNBase.build
E build: TypeError: list indices must be integers or slices, not str
看意思是list必须是整数,而不是str,但是我看了一下数组都是整数,到底是怎么回事

回复

使用道具 举报

1万

积分

18

威望

20

贡献

神级创客

Rank: 8Rank: 8

积分
10850

突出贡献

发表于 2021-8-26 10:49:40     
是不是输入节点和size不匹配
回复

使用道具 举报

46

积分

0

威望

0

贡献

技术小白

积分
46
发表于 2021-8-27 20:18:25     
zyk 发表于 2021-8-26 10:49
是不是输入节点和size不匹配

是的,解决了
回复

使用道具 举报

158

积分

0

威望

0

贡献

技术小白

积分
158
发表于 2022-4-25 20:42:34     
zyk 发表于 2021-8-26 10:49
是不是输入节点和size不匹配

我与遇到了类似的问题,在X86设备上跑RKNN_TOOKIT2的模型转换demo(rknn-toolkit2/examples/onnx/yolov5)时,运行test.py文件出现一下报错
  1. E build: Catch exception when building RKNN model!
  2. E build: Traceback (most recent call last):
  3. E build:   File "rknn/api/rknn_base.py", line 1590, in rknn.api.rknn_base.RKNNBase.build
  4. E build:   File "rknn/api/rknn_base.py", line 344, in rknn.api.rknn_base.RKNNBase._generate_rknn
  5. E build:   File "rknn/api/rknn_base.py", line 276, in rknn.api.rknn_base.RKNNBase._export_rknn
  6. E build: ImportError: libpython3.6m.so.1.0: cannot open shared object file: No such file or directory
  7. Build model failed!
复制代码


看起来是缺少库,但是我find了一下,这个文件是存在的(/snap/gnome-3-34-1804/77/usr/lib/x86_64-linux-gnu/libpython3.6m.so.1.0
)就是没有识别,尝试移动到/usr/lib中也是会报错找不到库。请问这种情况怎么解决?
回复

使用道具 举报

1万

积分

18

威望

20

贡献

神级创客

Rank: 8Rank: 8

积分
10850

突出贡献

发表于 2022-4-26 14:44:34     
x86系统哪个版本的?目前使用的RKNN SDK是否为最新的版本
回复

使用道具 举报

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

本版积分规则

友情链接 : 爱板网 电子发烧友论坛 云汉电子社区 粤ICP备14022046号-2
快速回复 返回顶部 返回列表