Toybrick

转换tensorflow模型报错AttributeError: 'NoneType' object has no attribut...

reapp

注册会员

积分
60
QQ
发表于 2020-7-22 16:33:20    查看: 5769|回复: 2 | [复制链接]    打印 | 显示全部楼层
用tensorflow object detedction的模型进行转化的时候遇到问题
E Catch exception when loading tensorflow model: ./frozen_inference_graph.pb!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 188, in rknn.api.rknn_base.RKNNBase.load_tensorflow
E   File "rknn/base/RKNNlib/converter/convert_tf.py", line 90, in rknn.base.RKNNlib.converter.convert_tf.convert_tf.__init__
E   File "rknn/base/RKNNlib/converter/tensorflowloader.py", line 41, in rknn.base.RKNNlib.converter.tensorflowloader.TF_Graph_Preprocess.__init__
E AttributeError: 'NoneType' object has no attribute 'op'
done
--> Building model
Traceback (most recent call last):
  File "ssd.py", line 75, in <module>
    ret = rknn.build(do_quantization=True, dataset='./dataset.txt')
  File "/usr/local/lib/python3.5/dist-packages/rknn/api/rknn.py", line 220, in build
    inputs = self.rknn_base.net.get_input_layers()
AttributeError: 'NoneType' object has no attribute 'get_input_layers'


tensorflow版本是1.12, rknn是1.3.0


回复

使用道具 举报

leok

版主

积分
894
发表于 2020-7-22 17:23:21 | 显示全部楼层
tensorflow升级到1.13.0   模型也在该版本下训练保存。
rknn toolkit升级到1.3.2
回复

使用道具 举报

叽里咕噜象

注册会员

积分
90
发表于 2020-12-15 18:46:43 | 显示全部楼层
请问楼主解决了吗?

回复

使用道具 举报

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

本版积分规则

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


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