Toybrick

RKNN转换出错

awiueret

注册会员

积分
124
楼主
发表于 2021-10-5 17:09:41    查看: 3380|回复: 0 | [复制链接]    打印 | 只看该作者

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


回复

使用道具 举报

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

本版积分规则

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


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