Toybrick

标题: RKNN转换出错 [打印本页]

作者: awiueret    时间: 2021-10-5 17:09
标题: RKNN转换出错

主旨:tensorflow 轉rknn 出錯


eric@eric:~/RV1126_1109SDK/external/rknn-toolkit/examples/tensorflow/ssd_mobilenet_v1$ python3 ssd.py
--> Loading model
/home/eric/.local/lib/python3.6/site-packages/onnx_tf/common/__init__.py:87: UserWarning: FrontendHandler.get_outputs_names is deprecated. It willr be removed in future release.. Use node.outputs instead.
  warnings.warn(message)
E Catch exception when loading tensorflow model: ./face_mask_detection.pb!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 221, in rknn.api.rknn_base.RKNNBase.load_tensorflow
E   File "rknn/base/RKNNlib/converter/convert_tf.py", line 93, in rknn.base.RKNNlib.converter.convert_tf.convert_tf.__init__
E   File "rknn/base/RKNNlib/converter/tensorflowloader.py", line 38, in rknn.base.RKNNlib.converter.tensorflowloader.TF_Graph_Preprocess.__init__
E   File "rknn/base/RKNNlib/converter/tensorflowloader.py", line 54, in rknn.base.RKNNlib.converter.tensorflowloader.TF_Graph_Preprocess.scan_and_optim_graph
E   File "/home/eric/.local/lib/python3.6/site-packages/tensorflow/python/framework/graph_util_impl.py", line 174, in extract_sub_graph
E     _assert_nodes_are_present(name_to_node, dest_nodes)
E   File "/home/eric/.local/lib/python3.6/site-packages/tensorflow/python/framework/graph_util_impl.py", line 133, in _assert_nodes_are_present
E     assert d in name_to_node, "%s is not in graph" % d
E AssertionError: concat_1 is not in graph
done
--> Building model
Traceback (most recent call last):
  File "ssd.py", line 85, in <module>
    rknn.build(do_quantization=True, dataset='./dataset.txt')
  File "/home/eric/.local/lib/python3.6/site-packages/rknn/api/rknn.py", line 231, in build
    inputs = self.rknn_base.net.get_input_layers()
AttributeError: 'NoneType' object has no attribute 'get_input_layers'

===============================================================================
配置環境:
tensorboard                   1.11.0
tensorboard-data-server       0.6.1
tensorboard-plugin-wit        1.8.0
tensorflow                    1.11.0
tensorflow-estimator          1.13.0
tensorflow-gpu                1.11.0
termcolor                     1.1.0
torch                         1.2.0
torchvision                   0.4.0
macaroonbakery                1.1.3
Mako                          1.0.7
Markdown                      3.3.4
MarkupSafe                    2.0.1
matplotlib                    3.3.4
mock                          4.0.3
mxnet                         1.5.0
netifaces                     0.10.4
networkx                      1.11
numpy                         1.19.5
oauth                         1.0.1
oauthlib                      3.1.0
olefile                       0.45.1
onboard                       1.4.1







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