简体   繁体   English

无法将张量流模型转换为tflite

[英]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.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM