Toybrick

标题: tensorflow版的yolov2 在模型转换环节出错 [打印本页]

作者: lynya    时间: 2019-4-16 18:42
标题: tensorflow版的yolov2 在模型转换环节出错
  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版本的问题还是训练代码的问题。





欢迎光临 Toybrick (https://t.rock-chips.com/) Powered by Discuz! X3.3