zwmasdf 发表于 2021-8-24 13:15:48

rknn-toolkit2模型转化失败

import numpy as np
import re
import math
import random
import cv2
from rknn.api import RKNN
rknn = RKNN()
rknn.config(mean_values=, std_values=)
rknn.load_tensorflow(tf_pb='./foods26_frozen.pb',
                     inputs=['input'],
                     outputs=['InceptionV3/Predictions/Reshape_1'],
                     input_size_list=[])
rknn.build(do_quantization=True, dataset='./dataset.txt')
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，但是我看了一下数组都是整数，到底是怎么回事

zyk 发表于 2021-8-26 10:49:40

是不是输入节点和size不匹配

zwmasdf 发表于 2021-8-27 20:18:25

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

是的，解决了

lozenyin 发表于 2022-4-25 20:42:34

zyk 发表于 2021-8-26 10:49
是不是输入节点和size不匹配
我与遇到了类似的问题，在X86设备上跑RKNN_TOOKIT2的模型转换demo（rknn-toolkit2/examples/onnx/yolov5）时，运行test.py文件出现一下报错
E build: Catch exception when building RKNN model!
E build: Traceback (most recent call last):
E build:   File "rknn/api/rknn_base.py", line 1590, in rknn.api.rknn_base.RKNNBase.build
E build:   File "rknn/api/rknn_base.py", line 344, in rknn.api.rknn_base.RKNNBase._generate_rknn
E build:   File "rknn/api/rknn_base.py", line 276, in rknn.api.rknn_base.RKNNBase._export_rknn
E build: ImportError: libpython3.6m.so.1.0: cannot open shared object file: No such file or directory
Build model failed!


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

zyk 发表于 2022-4-26 14:44:34

x86系统哪个版本的？目前使用的RKNN SDK是否为最新的版本
页: [1]
查看完整版本: rknn-toolkit2模型转化失败