rknn-toolkit支持Tensorflow v1.13.1

[复制链接] · 发表于 2019-5-10 15:57:52

请教下使用的是rknn-toolkit v0.9.9，安装了tensorflow-1.10.1-cp36-cp36m-linux_aarch64，在转换模型的时候提示版本问题，请问下怎么升级到更高版本的tensorflow_aarch64，例如1.13.1？？模型使用的api在1.10.1上有bug，谢谢

只看该作者 · 发表于 2019-5-10 17:52:30

aarch64 可以自己下载tensorflow源码在板子上编译下，比较麻烦。
最简单的方法建议你在pc上装了1.13.1和rknn后在pc端转换。。

只看该作者 · 发表于 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么??

只看该作者 · 发表于 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 ...

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