张量的输入大小与张量的输出大小不匹配
我已经在命令提示符下执行了以下操作
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)。
答案 0 :(得分:0)
pip install --upgrade tensorflow
。
这对我有用,我已经成功地将.pb转换为.tflite