|
主旨: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
|
|