Toybrick

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

Deeelta

新手上路

积分
13
楼主
发表于 2020-5-16 19:58:40    查看: 18398|回复: 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'

回复

使用道具 举报

Deeelta

新手上路

积分
13
沙发
 楼主| 发表于 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是可用了吗
回复

使用道具 举报

Deeelta

新手上路

积分
13
板凳
 楼主| 发表于 2020-5-18 11:24:30 | 显示全部楼层
jefferyzhang 发表于 2020-5-18 10:09
不可以的,我们还没做好,没有对外发布

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

使用道具 举报

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

本版积分规则

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


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