[英]Unable to convert the tensorflow model to tflite
Input size of the tensor does not matching the output size of the tensor 张量的输入大小与张量的输出大小不匹配
I have executed the following in command prompt 我已经在命令提示符下执行了以下命令
tflite_convert /
--output_file=C:\Users\TS-PC04\Desktop\toutput.tflite /
--graph_def_file=C:\Users\TS-PC04\Desktop\t.pb /
--input_shape=1,513,513,3 /
--input_arrays=ImageTensor /
--output_arrays=SemanticPredictions
The error I am getting is Check failed: input_shape.dims().size() == op->size.size() (4 vs. 3). 我得到的错误是检查失败:input_shape.dims()。size()== op-> size.size()(4比3)。
pip install --upgrade tensorflow
. pip install --upgrade tensorflow
This works for me and I am successfully converting .pb to .tflite 这对我有效,并且我已成功将.pb转换为.tflite
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.