Toybrick

tensorflow版的yolov2 在模型转换环节出错

lynya

新手上路

积分
21
楼主
发表于 2019-4-16 18:42:09    查看: 8614|回复: 0 | [复制链接]    打印 | 只看该作者
  1. D Try match SpaceToDepth high_res_reshape
  2. W Not match node high_res_reshape SpaceToDepth
  3. E Catch exception when loading tensorflow model: ./yolov2.pb!
  4. T Traceback (most recent call last):
  5. T   File "rknn/api/rknn_base.py", line 180, in rknn.api.rknn_base.RKNNBase.load_
  6. tensorflow
  7. T   File "rknn/base/RKNNlib/converter/convert_tf.py", line 565, in rknn.base.RKN
  8. Nlib.converter.convert_tf.convert_tf.match_paragraph_and_param
  9. T   File "rknn/base/RKNNlib/converter/convert_tf.py", line 464, in rknn.base.RKN
  10. Nlib.converter.convert_tf.convert_tf._tf_push_ready_node
  11. T TypeError: 'NoneType' object is not iterable
  12. done
  13. --> Building model
  14. E Catch exception when building RKNN model!
复制代码
如上,tensorflow版本是1.10.1,有一行代码是“high_res = tf.space_to_depth(high_res, 2, name="high_res_reshape")”,在模型转换时报错了,我理解这里RKNN里有一个名称“ SpaceToDepth”的函数,但是在匹配的时候出错了。请问有人知道这是什么原因导致的吗?我看论坛上yolov2应该是一个已经得到大家测试过的模型,不知道是我这里出错是tensorflow版本的问题还是训练代码的问题。
回复

使用道具 举报

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

本版积分规则

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


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