|
发表于 2021-2-3 15:58:01
|
显示全部楼层
yolov3 的tensorflow pb模型 转 rknn模型报了以下错误 代码用的 rknn_convert 中提供的代码
--> Loading model...
C:\ProgramData\Anaconda3\envs\rk\lib\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)
E Catch exception when loading tensorflow model: D:\BaiduNetdiskDownload\RKNN-Toolkit\rknn-toolkit-v1.4.0\examples\rknn_convert\models\tensorflow\mobilenet-ssd\output_graph.pb!
E Traceback (most recent call last):
E File "C:\ProgramData\Anaconda3\envs\rk\lib\site-packages\tensorflow\python\framework\importer.py", line 418, in import_graph_def
E graph._c_graph, serialized, options) # pylint: disable=protected-access
E tensorflow.python.framework.errors_impl.InvalidArgumentError: Input 0 of node IteratorGetNext was passed float from Iterator_out_0_placeholder:0 incompatible with expected resource.
E During handling of the above exception, another exception occurred:
E Traceback (most recent call last):
E File "rknn\api\rknn_base.py", line 222, in rknn.api.rknn_base.RKNNBase.load_tensorflow
E File "rknn\base\RKNNlib\converter\convert_tf.py", line 530, in rknn.base.RKNNlib.converter.convert_tf.convert_tf.pre_process
E File "rknn\base\RKNNlib\converter\tensorflowloader.py", line 91, in rknn.base.RKNNlib.converter.tensorflowloader.TF_Graph_Preprocess.pre_proces
E File "rknn\base\RKNNlib\converter\tensorflowloader.py", line 809, in rknn.base.RKNNlib.converter.tensorflowloader.TF_Graph_Preprocess.calc_2_const
E File "rknn\base\RKNNlib\converter\tf_util.py", line 371, in rknn.base.RKNNlib.converter.tf_util.TFProto_Util.query_tensor
E File "rknn\base\RKNNlib\converter\tf_util.py", line 372, in rknn.base.RKNNlib.converter.tf_util.TFProto_Util.query_tensor
E File "C:\ProgramData\Anaconda3\envs\rk\lib\site-packages\tensorflow\python\util\deprecation.py", line 488, in new_func
E return func(*args, **kwargs)
E File "C:\ProgramData\Anaconda3\envs\rk\lib\site-packages\tensorflow\python\framework\importer.py", line 422, in import_graph_def
E raise ValueError(str(e))
E ValueError: Input 0 of node IteratorGetNext was passed float from Iterator_out_0_placeholder:0 incompatible with expected resource.
Traceback (most recent call last):
File "D:/BaiduNetdiskDownload/RKNN-Toolkit/rknn-toolkit-v1.4.0/examples/rknn_convert/rknn_convert.py", line 89, in <module>
convert_model(model_path, out_path, pre_compile)
File "D:/BaiduNetdiskDownload/RKNN-Toolkit/rknn-toolkit-v1.4.0/examples/rknn_convert/rknn_convert.py", line 72, in convert_model
rknn.build(do_quantization=model['quantize'], dataset=dataset_path, pre_compile=pre_compile)
File "C:\ProgramData\Anaconda3\envs\rk\lib\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'
done
--> Build RKNN model...
Process finished with exit code 1
请问可能是什么问题啊 |
|