|
在用TF版本ssd-mobilenetv1训练后,转rknn模型时出错:AttributeError: 'NoneType' object has no attribute 'get_input_layers'. 网络输入尺寸300x300x3,类别数为2. tf版本为1.13.1,rknn-toolkit版本为1.3.0. 平台为:Ubuntu16.04虚拟机.
log日志如下:
python ssd_person.py
WARNING: The TensorFlow contrib module will not be included in TensorFlow 2.0.
For more information, please see:
* https://github.com/tensorflow/co ... 7-contrib-sunset.md
* https://github.com/tensorflow/addons
If you depend on functionality not listed there, please file an issue.
/home/devin/.local/lib/python3.5/site-packages/onnx_tf/common/__init__.py:87: UserWarning: FrontendHandler.get_outputs_names is deprecated. It will be removed in future release.. Use node.outputs instead.
warnings.warn(message)
W verbose file path is invalid, debug info will not dump to file.
--> Loading model
D import clients finished
W:tensorflow:From /home/devin/.local/lib/python3.5/site-packages/rknn/api/rknn.py:68: extract_sub_graph (from tensorflow.python.framework.graph_util_impl) is deprecated and will be removed in a future version.
Instructions for updating:
Use tf.compat.v1.graph_util.extract_sub_graph
E Catch exception when loading tensorflow model: /mnt/4667845a-b9d8-4961-8c27-6796aba9e29a/rknn-toolkit-V1.2.0/example/mobilenet-ssd/frozen_inference_graph.pb!
T Traceback (most recent call last):
T File "rknn/api/rknn_base.py", line 135, in rknn.api.rknn_base.RKNNBase.load_tensorflow
T File "rknn/base/RKNNlib/converter/convert_tf.py", line 107, in rknn.base.RKNNlib.converter.convert_tf.convert_tf.__init__
T File "rknn/base/RKNNlib/converter/tensorflowloader.py", line 53, in rknn.base.RKNNlib.converter.tensorflowloader.TF_Graph_Preprocess.__init__
T AttributeError: 'NoneType' object has no attribute 'op'
done
--> Building model
Traceback (most recent call last):
File "ssd_person.py", line 77, in <module>
rknn.build(do_quantization=False)
File "/home/devin/.local/lib/python3.5/site-packages/rknn/api/rknn.py", line 163, in build
inputs = self.rknn_base.net.get_input_layers()
AttributeError: 'NoneType' object has no attribute 'get_input_layers'
|
|