错误地创建了tflight图

问题描述 投票:0回答:1

我有以下冻结推理图。这是使用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)的谷歌图形的图形?为了给您一个更清晰的问题,请参见下图enter image description here

左边的图是我的tflite图,右边的图是google的deeplab图。如何获得与右图类似的结果?

python tensorflow machine-learning computer-vision artificial-intelligence
1个回答
0
投票
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

这解决了我的问题

© www.soinside.com 2019 - 2024. All rights reserved.