错误创建的tflight图

Question

我有以下冻结的推理图。这用于使用Deeplab（download graph here）进行语义分割。我将此图转换为tflite格式

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

转换后，图形如下（download it here） 我的问题是如何获取类似于可用的Deeplab（here）的googles图？为了给您一个更清晰的问题，请参见下图

左图是我的tflite图，右图是Google的deeplab图。如何获得类似于右侧图的结果？

Answer 1

tflite_convert \
  --output_file=test2.lite \
  --graph_def_file=frozen_inference_graph_3mbvoc.pb \
  --input_arrays=sub_2 \
  --output_arrays=ResizeBilinear_2 \
  --input_shapes=1,450,600,3 \
  --inference_input_type=QUANTIZED_UINT8 \
  --inference_type=FLOAT \
  --mean_values=128 \
  --std_dev_values=128

这解决了我的问题