Toybrick

rknn-toolkit支持Tensorflow v1.13.1

jefby

新手上路

积分
16
楼主
发表于 2019-5-10 15:57:52    查看: 9549|回复: 3 | [复制链接]    打印 | 只看该作者
请教下使用的是rknn-toolkit v0.9.9,安装了tensorflow-1.10.1-cp36-cp36m-linux_aarch64,在转换模型的时候提示版本问题,请问下怎么升级到更高版本的tensorflow_aarch64,例如1.13.1??模型使用的api在1.10.1上有bug,谢谢
回复

使用道具 举报

jefferyzhang

版主

积分
13578
沙发
发表于 2019-5-10 17:52:30 | 只看该作者
aarch64 可以自己下载tensorflow源码在板子上编译下,比较麻烦。
最简单的方法建议你在pc上装了1.13.1和rknn后在pc端转换。。
回复

使用道具 举报

jefby

新手上路

积分
16
板凳
 楼主| 发表于 2019-5-12 15:40:34 | 只看该作者
本帖最后由 jefby 于 2019-5-12 15:54 编辑
jefferyzhang 发表于 2019-5-10 17:52
aarch64 可以自己下载tensorflow源码在板子上编译下,比较麻烦。
最简单的方法建议你在pc上装了1.13.1和rkn ...

我尝试用第一种方法, 在PC上安装tensorflow 1.13.1, 然后使用x86_64下的rknn-toolkit 1.0.0 转换tensorflow pb模型为rknn格式, 碰到了如下错误:

--> Loading model
W Not match node select_bboxes/map/TensorArrayStack_1/TensorArrayGatherV3 TensorArrayGatherV3
E Catch exception when loading tensorflow model: ./rsu_perception_0402.pb!
T Traceback (most recent call last):
T   File "rknn/api/rknn_base.py", line 185, in rknn.api.rknn_base.RKNNBase.load_tensorflow
T   File "rknn/base/RKNNlib/converter/convert_tf.py", line 589, in rknn.base.RKNNlib.converter.convert_tf.convert_tf.match_paragraph_and_param
T   File "rknn/base/RKNNlib/converter/convert_tf.py", line 488, in rknn.base.RKNNlib.converter.convert_tf.convert_tf._tf_push_ready_node
T TypeError: 'NoneType' object is not iterable
done

是现在不支持ops TensorArrayGatherV3么??
回复

使用道具 举报

jefferyzhang

版主

积分
13578
地板
发表于 2019-5-13 08:28:17 | 只看该作者
jefby 发表于 2019-5-12 15:40
我尝试用第一种方法, 在PC上安装tensorflow 1.13.1, 然后使用x86_64下的rknn-toolkit 1.0.0 转换tensorflo ...

看过去是这样,你可以试着把这层去掉再转换下能不能成功
回复

使用道具 举报

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

本版积分规则

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


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