pip安装的rknn-toolkit2导入tensorflow的pb模型时 AttributeError: module 'rknn' has no attribute 'load_tensorflow'

2024-10-30 13:02

(base) huangzr@LAPTOP-ET7COTQM:~$ ls
Fairy.pb Fairy.pb:Zone.Identifier anaconda3 contact.py luckfox_pico_rknn_example tensorflow-demo
(base) huangzr@LAPTOP-ET7COTQM:~$ python contact.py
Traceback (most recent call last):
File "/home/huangzr/contact.py", line 5, in <module>
ret = rknn.load_tensorflow(tf_pb='/home/huangzr/Fairy.pb',
^^^^^^^^^^^^^^^^^^^^
AttributeError: module 'rknn' has no attribute 'load_tensorflow'

contact.py:
import rknn

from rknn.api import RKNN

ret = rknn.load_tensorflow(tf_pb='/home/huangzr/Fairy.pb',
inputs=['Preprocessor/sub'],
outputs=['concat', 'concat_1'],
input_size_list=[[None, 280, 210, 3]])

2024-10-31 2:25

您好，您的代码中提示了module 'rknn' has no attribute 'load_tensorflow' 说明调用的rknn模块没有load_tensorflow这个函数，但是在03_Rockchip_RKNPU_API_Reference_RKNN_Toolkit2_V1.6.0_CN.pdf的说明中有相关的说明，建议更新一下rknn-toolkit2 的库（仓库地址：https://github.com/airockchip/rknn-toolkit2）。
同时建议先将tensorflow模型转换为 ONNX 模型后再进行模型导入，rknn-toolkit2 对 ONNX 模型的支持性更好，再提示报错时也可以方便找到是哪一个神经网络层出现了问题

2024-10-31 4:31

我是直接pip安装的

2024-10-31 4:31

我只有tensorflow的pb模型

2024-10-31 6:02

您提供的日志中可以看出您是有类似conda的python虚拟环境的，建议创建一个conda环境重新安装https://github.com/airockchip/rknn-tool ... 2/packages 中的工具包的依赖环境，我实测过rknn.api 中是有 load_tensorflow 函数的

Code: Select all

from rknn.api import RKNN

rknn = RKNN(verbose=False)
rknn.load_tensorflow() #直接调用回报错missing 4 required positional arguments 但是不是提示module 'rknn' has no attribute 'load_tensorflow'

2024-11-09 4:15

Traceback (most recent call last):
File "contact.py", line 8, in <module>
ret = RKNN.load_tensorflow(tf_pb='/home/huangzr/Fairy.pb',
TypeError: load_tensorflow() missing 1 required positional argument: 'self'
现在变成了这样，文档上没有self参数

2024-11-11 1:18

请检查是否有转换代码中是有 rknn=RKNN() 这样的类初始化代码，如果确定有进行类初始化尝试将rknn传入rknn.load_tensorflow第一个参数，以上方法都无法解决请在https://github.com/rockchip-linux/rknn-toolkit2向Rockchip咨询，rknn-toolkit2 我们没有参与开发和维护，设计细节了解有限。