准备张量分配时出现意外故障

Question

我正在使用tflite进行语义分割。我有一个经过训练的模型，可以从背景中分割对象，这个模型是在深度实验室上训练的。

我已经使用下面的代码将这个模型(冻结的推理图)转换为tflite格式:

tflite_convert \
  --output_file=test.lite \
  --graph_def_file=frozen_inference_graph.pb \
  --input_arrays=ImageTensor \
  --output_arrays=SemanticPredictions \
  --input_shapes=1,600,450,3 \
  --inference_input_type=QUANTIZED_UINT8 \
  --inference_type=FLOAT \
  --mean_values=128 \
  --std_dev_values=128 

模型在Android上加载，但是当我尝试运行推理时，它会给我这个错误：

原因：java.lang.IllegalStateException:内部错误：准备张量分配时出现意外失败：third_party/tensorflow/lite/kernels/unpack。cc:54数字尺寸（输入）

1不是真的。4号节点(解包)准备失败。

如何解决此错误？

Answer 1

它适用于以下命令：

bazel-bin/tensorflow/lite/toco/toco \
  --input_file=deeplabv3_mnv2_pascal_tain.pb \
  --output_file=test.tflite \
  --inference_input_type=QUANTIZED_UINT8  \
  --inference_type=FLOAT \
  --input_arrays=ImageTensor  \
  --output_arrays=SemanticPredictions  \
  --input_shapes=1,513,513,3 \
  --mean_values=128 \
  --std_dev_values=128

当我从源头安装张量流时，它起作用了。为了从我使用的源安装张量流（链接）