tensorflow的pb模型转rknn报错model or data is None

[复制链接] · 发表于 2019-9-17 15:24:42

1、已经生成了tensorflow的pb模型，并且在ubuntu系统中已经验证过这个模型了2、在TB-RK3399Pro开发板上转换模型时，参照的是官方的wiki教程，但是运行脚本时，出现以下报错：
[toybrick@localhost mnist_test]$ python3.6 mnist_rknn.py
/usr/lib64/python3.6/site-packages/h5py/__init__.py:36: FutureWarning: Conversion of the second argument of issubdtype from `float` to `np.floating` is deprecated. In future, it will be treated as `np.float64 == np.dtype(float).type`.
from ._conv import register_converters as _register_converters
load tensorflow model
E Catch exception when loading tensorflow model: /home/toybrick/tensorflow-ai/mnist_test/grf.pb!
T Traceback (most recent call last):
T File "rknn/api/rknn_base.py", line 136, in rknn.api.rknn_base.RKNNBase.load_tensorflow
T File "rknn/base/RKNNlib/converter/convert_tf.py", line 101, in rknn.base.RKNNlib.converter.convert_tf.convert_tf.__init__
T TypeError: can only concatenate list (not "int") to list
load tensorflow model end
build rknn model
E Model or data is None, please load model first.
build rknn model end
E RKNN model data is None, please load model first!
其中红色的代码，是自己程序里面写的print信息。请问是什么原因造成这样的报错。

只看该作者 · 发表于 2019-9-17 18:10:55

T TypeError: can only concatenate list (not "int") to list
你验证过你的pb么，你这个看过去是keras转pb，但是pb是不是有问题？

只看该作者 · 发表于 2019-9-18 08:50:49

jefferyzhang 发表于 2019-9-17 18:10
T TypeError: can only concatenate list (not "int") to list
你验证过你的pb么，你这个看过去是keras转pb ...

pb验证过了，用的是tf.gfile.GFile保存的pb模型

只看该作者 · 发表于 2019-9-18 11:13:34

verbose打开。
如果是mnist肯定是能转的。检查下代码哪有问题。
我们也会出个教程转mnist的

只看该作者 · 发表于 2019-9-20 16:50:59

jefferyzhang 发表于 2019-9-18 11:13
verbose打开。
如果是mnist肯定是能转的。检查下代码哪有问题。
我们也会出个教程转mnist的 ...

你好请问官方API中，File "rknn/base/RKNNlib/converter/convert_tf.py", line 101, in rknn.base.RKNNlib.converter.convert_tf.convert_tf.__init__这个步骤具体做什么工作，有源码可以查看一下吗

只看该作者 · 发表于 2019-9-20 18:07:28

mnist有问题请参看我们新出的mnist教程：http://t.rock-chips.com/forum.ph ... &extra=page%3D2

你说的这代码干什么我们也不清楚，这是NPU部门做的，一般就是OP的转换工作而已。