Toybrick

tensorflow keras模型转rknn报错,如何解决?

Deeelta

新手上路

积分
13
楼主
发表于 2020-5-16 19:58:40    查看: 27784|回复: 10 | [复制链接]    打印 | 只看该作者
模型是h5,使用了load_keras这个api, rknn.load_keras(model=H5_PATH, inputs=['input_1_1'], input_size_list='None,416,416,3', outputs=['conv2d_59_1/BiasAdd', 'conv2d_67_1/BiasAdd', 'conv2d_75_1/BiasAdd']),报了如下错误:

E Catch exception when loading keras model: ./model/kitchen_model_v5.h5!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 542, in rknn.api.rknn_base.RKNNBase.load_keras
E   File "rknn/base/RKNNlib/app/importer/import_keras.py", line 179, in rknn.base.RKNNlib.app.importer.import_keras.ImportKeras.run
E AttributeError: 'list' object has no attribute 'split'

回复

使用道具 举报

jefferyzhang

版主

积分
13576
沙发
发表于 2020-5-17 10:37:50 | 只看该作者
tensorflow版本必须是1.13.1,
rknn-toolkit版本最好用最新的1.3.2,
我们对外都还没有发布load_keras的接口,你rknn-toolkit用的是啥版本
回复

使用道具 举报

AI_liu

注册会员

积分
138
板凳
发表于 2020-5-17 11:44:11 | 只看该作者
jefferyzhang 发表于 2020-5-17 10:37
tensorflow版本必须是1.13.1,
rknn-toolkit版本最好用最新的1.3.2,
我们对外都还没有发布load_keras的接 ...

rknn-toolkit版本最好用最新的1.3.2版本你们放出来了吗?
回复

使用道具 举报

jefferyzhang

版主

积分
13576
地板
发表于 2020-5-17 11:58:07 | 只看该作者
AI_liu 发表于 2020-5-17 11:44
rknn-toolkit版本最好用最新的1.3.2版本你们放出来了吗?

toolkit放出来了,toybrick板子可以直接通过pip3升级,或者从这里下载:https://github.com/rockchip-linux/rknn-toolkit
drv还在测试流程中
就算是1.3.2,也还没支持keras啊,你的load_keras哪里来的
回复

使用道具 举报

Deeelta

新手上路

积分
13
5#
 楼主| 发表于 2020-5-18 09:42:22 | 只看该作者
jefferyzhang 发表于 2020-5-17 10:37
tensorflow版本必须是1.13.1,
rknn-toolkit版本最好用最新的1.3.2,
我们对外都还没有发布load_keras的接 ...

tensorflow是1.11.0, rknn是1.3.0的,load_keras的接口是调试的时候发现的~
那1.3.2这个版本的load_keras是可用了吗
回复

使用道具 举报

jefferyzhang

版主

积分
13576
6#
发表于 2020-5-18 10:09:04 | 只看该作者
Deeelta 发表于 2020-5-18 09:42
tensorflow是1.11.0, rknn是1.3.0的,load_keras的接口是调试的时候发现的~
那1.3.2这个版本的load_keras ...

不可以的,我们还没做好,没有对外发布
回复

使用道具 举报

Deeelta

新手上路

积分
13
7#
 楼主| 发表于 2020-5-18 11:24:30 | 只看该作者
jefferyzhang 发表于 2020-5-18 10:09
不可以的,我们还没做好,没有对外发布

有版本计划吗 现在搞一个项目想尝试加载keras
回复

使用道具 举报

jefferyzhang

版主

积分
13576
8#
发表于 2020-5-18 11:26:13 | 只看该作者
Deeelta 发表于 2020-5-18 11:24
有版本计划吗 现在搞一个项目想尝试加载keras

这个我是不知道的,我不是NPU部门的,他们发布时候会通知我,但是不会跟我说他们计划的。。
回复

使用道具 举报

wen0554

新手上路

积分
26
9#
发表于 2020-9-25 15:40:48 | 只看该作者
你的参数写错了,应该是:
rknn.load_keras(model=H5_PATH, inputs='input_1_1', input_size_list='416,416,3', outputs='conv2d_59_1/BiasAdd, conv2d_67_1/BiasAdd, conv2d_75_1/BiasAdd')

另外修改一下
lib/python3.6/site-packages/rknn/ap/rknn.py
回复

使用道具 举报

ztfmars

注册会员

积分
108
10#
发表于 2020-9-25 16:44:37 | 只看该作者
jefferyzhang 发表于 2020-5-18 11:26
这个我是不知道的,我不是NPU部门的,他们发布时候会通知我,但是不会跟我说他们计划的。。
...

1.4.0支持直接加载keras了么?
回复

使用道具 举报

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

本版积分规则

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


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