Toybrick

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

jandriux

注册会员

积分
90
发表于 2019-9-17 15:24:42    查看: 9523|回复: 5 | [复制链接]    打印 | 显示全部楼层
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信息。请问是什么原因造成这样的报错。


回复

使用道具 举报

jefferyzhang

版主

积分
13182
发表于 2019-9-17 18:10:55 | 显示全部楼层
T TypeError: can only concatenate list (not "int") to list
你验证过你的pb么,你这个看过去是keras转pb,但是pb是不是有问题?
回复

使用道具 举报

jandriux

注册会员

积分
90
 楼主| 发表于 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模型
回复

使用道具 举报

jefferyzhang

版主

积分
13182
发表于 2019-9-18 11:13:34 | 显示全部楼层
verbose打开。
如果是mnist肯定是能转的。检查下代码哪有问题。
我们也会出个教程转mnist的
回复

使用道具 举报

jandriux

注册会员

积分
90
 楼主| 发表于 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__这个步骤具体做什么工作,有源码可以查看一下吗
回复

使用道具 举报

jefferyzhang

版主

积分
13182
发表于 2019-9-20 18:07:28 | 显示全部楼层
mnist有问题请参看我们新出的mnist教程:http://t.rock-chips.com/forum.ph ... &extra=page%3D2

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

使用道具 举报

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

本版积分规则

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


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