Toybrick

标题: tensorflow keras模型转rknn报错，如何解决？ [打印本页]

作者: Deeelta 时间: 2020-5-16 19:58
标题: tensorflow keras模型转rknn报错，如何解决？
模型是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 时间: 2020-5-17 10:37
tensorflow版本必须是1.13.1，
rknn-toolkit版本最好用最新的1.3.2，
我们对外都还没有发布load_keras的接口，你rknn-toolkit用的是啥版本

作者: AI_liu 时间: 2020-5-17 11:44

jefferyzhang 发表于 2020-5-17 10:37
tensorflow版本必须是1.13.1，
rknn-toolkit版本最好用最新的1.3.2，
我们对外都还没有发布load_keras的接 ...

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

作者: jefferyzhang 时间: 2020-5-17 11:58

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 时间: 2020-5-18 09:42

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 时间: 2020-5-18 10:09

Deeelta 发表于 2020-5-18 09:42
tensorflow是1.11.0, rknn是1.3.0的，load_keras的接口是调试的时候发现的~
那1.3.2这个版本的load_keras ...

不可以的，我们还没做好，没有对外发布

作者: Deeelta 时间: 2020-5-18 11:24

jefferyzhang 发表于 2020-5-18 10:09
不可以的，我们还没做好，没有对外发布

有版本计划吗

现在搞一个项目想尝试加载keras

作者: jefferyzhang 时间: 2020-5-18 11:26

Deeelta 发表于 2020-5-18 11:24
有版本计划吗现在搞一个项目想尝试加载keras

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

作者: wen0554 时间: 2020-9-25 15:40
你的参数写错了，应该是：
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 时间: 2020-9-25 16:44

jefferyzhang 发表于 2020-5-18 11:26
这个我是不知道的，我不是NPU部门的，他们发布时候会通知我，但是不会跟我说他们计划的。。
...

1.4.0支持直接加载keras了么？

作者: jefferyzhang 时间: 2020-9-25 16:56

ztfmars 发表于 2020-9-25 16:44
1.4.0支持直接加载keras了么？

以发布的文档里头描述为准。

欢迎光临 Toybrick (https://t.rock-chips.com/)