Tensorflow版的yolov3（.pb）转rknn出现Exp不匹配问题

[复制链接] · 发表于 2019-11-1 16:58:37

报错如下：

log打印如下：
求解！

只看该作者 · 发表于 2019-11-1 17:00:56

verbose=True打开看下，如果是这个问题，那就是该OP不支持

只看该作者 · 发表于 2019-11-1 21:06:07

jefferyzhang 发表于 2019-11-1 17:00
verbose=True打开看下，如果是这个问题，那就是该OP不支持

verbose=True打开就是这样的
--- Logging error ---
Traceback (most recent call last):
  File "/usr/lib/python3.5/logging/__init__.py", line 982, in emit
stream.write(msg)
ValueError: I/O operation on closed file.
Call stack:
  File "tf2rknn.py", line 21, in <module>
ret = rknn.load_tensorflow(tf_pb = 'yolov3_coco.pb',inputs = ['input/input_data'],outputs = ['pred_lbbox/concat_2','pred_mbbox/concat_2','pred_sbbox/concat_2'],input_size_list = [[416,416,3]])
  File "/usr/local/lib/python3.5/dist-packages/rknn/api/rknn.py", line 68, in load_tensorflow
std_values=std_values)
  File "/home/yst/.local/lib/python3.5/site-packages/tensorflow/python/util/deprecation_wrapper.py", line 119, in __getattr__
_call_location(), full_name, rename)
  File "/home/yst/.local/lib/python3.5/site-packages/tensorflow/python/platform/tf_logging.py", line 166, in warning
get_logger().warning(msg, *args, **kwargs)
Message: 'From %s: The name %s is deprecated. Please use %s instead.\n'
Arguments: ('/usr/local/lib/python3.5/dist-packages/rknn/api/rknn.py:68', 'tf.GraphDef', 'tf.compat.v1.GraphDef')
--- Logging error ---
Traceback (most recent call last):
  File "/usr/lib/python3.5/logging/__init__.py", line 982, in emit
stream.write(msg)
ValueError: I/O operation on closed file.
Call stack:
  File "tf2rknn.py", line 21, in <module>
ret = rknn.load_tensorflow(tf_pb = 'yolov3_coco.pb',inputs = ['input/input_data'],outputs = ['pred_lbbox/concat_2','pred_mbbox/concat_2','pred_sbbox/concat_2'],input_size_list = [[416,416,3]])
  File "/usr/local/lib/python3.5/dist-packages/rknn/api/rknn.py", line 68, in load_tensorflow
std_values=std_values)
  File "/home/yst/.local/lib/python3.5/site-packages/tensorflow/python/util/deprecation.py", line 323, in new_func
instructions)
  File "/home/yst/.local/lib/python3.5/site-packages/tensorflow/python/platform/tf_logging.py", line 166, in warning
get_logger().warning(msg, *args, **kwargs)
Message: 'From %s: %s (from %s) is deprecated and will be removed %s.\nInstructions for updating:\n%s'
Arguments: ('/usr/local/lib/python3.5/dist-packages/rknn/api/rknn.py:68', 'extract_sub_graph', 'tensorflow.python.framework.graph_util_impl', 'in a future version', 'Use `tf.compat.v1.graph_util.extract_sub_graph`')
--- Logging error ---
Traceback (most recent call last):
  File "/usr/lib/python3.5/logging/__init__.py", line 982, in emit
stream.write(msg)
ValueError: I/O operation on closed file.
Call stack:
  File "tf2rknn.py", line 21, in <module>
ret = rknn.load_tensorflow(tf_pb = 'yolov3_coco.pb',inputs = ['input/input_data'],outputs = ['pred_lbbox/concat_2','pred_mbbox/concat_2','pred_sbbox/concat_2'],input_size_list = [[416,416,3]])
  File "/usr/local/lib/python3.5/dist-packages/rknn/api/rknn.py", line 68, in load_tensorflow
std_values=std_values)
  File "/home/yst/.local/lib/python3.5/site-packages/tensorflow/python/util/deprecation_wrapper.py", line 119, in __getattr__
_call_location(), full_name, rename)
  File "/home/yst/.local/lib/python3.5/site-packages/tensorflow/python/platform/tf_logging.py", line 166, in warning
get_logger().warning(msg, *args, **kwargs)
Message: 'From %s: The name %s is deprecated. Please use %s instead.\n'
Arguments: ('/usr/local/lib/python3.5/dist-packages/rknn/api/rknn.py:68', 'tf.Session', 'tf.compat.v1.Session')
--- Logging error ---
Traceback (most recent call last):
  File "/usr/lib/python3.5/logging/__init__.py", line 982, in emit
stream.write(msg)
ValueError: I/O operation on closed file.
Call stack:
  File "tf2rknn.py", line 21, in <module>
ret = rknn.load_tensorflow(tf_pb = 'yolov3_coco.pb',inputs = ['input/input_data'],outputs = ['pred_lbbox/concat_2','pred_mbbox/concat_2','pred_sbbox/concat_2'],input_size_list = [[416,416,3]])
  File "/usr/local/lib/python3.5/dist-packages/rknn/api/rknn.py", line 68, in load_tensorflow
std_values=std_values)
  File "/home/yst/.local/lib/python3.5/site-packages/tensorflow/python/util/deprecation_wrapper.py", line 119, in __getattr__
_call_location(), full_name, rename)
  File "/home/yst/.local/lib/python3.5/site-packages/tensorflow/python/platform/tf_logging.py", line 166, in warning
get_logger().warning(msg, *args, **kwargs)
Message: 'From %s: The name %s is deprecated. Please use %s instead.\n'
Arguments: ('/usr/local/lib/python3.5/dist-packages/rknn/api/rknn.py:68', 'tf.global_variables_initializer', 'tf.compat.v1.global_variables_initializer')
W Not match node pred_lbbox/Exp Exp
E Catch exception when loading tensorflow model: yolov3_coco.pb!
T Traceback (most recent call last):
T File "rknn/api/rknn_base.py", line 186, in rknn.api.rknn_base.RKNNBase.load_tensorflow
T File "rknn/base/RKNNlib/converter/convert_tf.py", line 589, in rknn.base.RKNNlib.converter.convert_tf.convert_tf.match_paragraph_and_param
T File "rknn/base/RKNNlib/converter/convert_tf.py", line 488, in rknn.base.RKNNlib.converter.convert_tf.convert_tf._tf_push_ready_node
T TypeError: 'NoneType' object is not iterable
Load model failed!

只看该作者 · 发表于 2019-11-1 22:45:12

W Not match node pred_lbbox/Exp Exp
是这个op不支持，你可以使用自定义op自己写一个

只看该作者 · 发表于 2019-11-3 11:12:36

jefferyzhang 发表于 2019-11-1 22:45
W Not match node pred_lbbox/Exp Exp
是这个op不支持，你可以使用自定义op自己写一个 ...

请问，op是啥？具体流程怎么搞呢

Tensorflow版的yolov3（.pb）转rknn出现Exp不匹配问题

本帖子中包含更多资源