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

时间:2019-06-17 12:56:44

标签: python-3.x tensorflow tensorflow-lite

张量的输入大小与张量的输出大小不匹配

我已经在命令提示符下执行了以下操作

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

我得到的错误是 检查失败:input_shape.dims()。size()== op-> size.size()(4比3)。

1 个答案:

答案 0 :(得分:0)

pip install --upgrade tensorflow
这对我有用,我已经成功地将.pb转换为.tflite