错误创建的tflight图

问题描述

我有以下冻结的推理图.这是使用Deeplab进行语义分割的方法(在此处下载图).我将此图转换为tflite格式

I have the following frozen inference graph. This is for semantic segmentation using Deeplab (download graph here). I converting this graph to tflite format

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

转换后的图形如下(在此处下载)我的问题是如何获取类似于可用的deeplab的Google谷歌图表(

After conversion the graph looks as follows(download it here)My question is how do I obtain graph similar to googles graph of deeplab available (

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

the graph on left is my tflite graph and the graph on right is graph of deeplab by google. How do I obtain results similar to graph on right?

推荐答案

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

这解决了我的问题

这篇关于错误创建的tflight图的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！