我有一个如下所示的冻结推理图。这是用于语义分割的Deeplab模型(下载图形)。我正在将这个图形转换为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
转换后的图形如下所示(下载)。我的问题是如何获得类似于谷歌提供的Deeplab图形(这里)?为了让我的问题更清晰,请看下面的图片
左图是我的tflite图形,右图是谷歌的Deeplab图形。我如何获得与右图类似的结果?
回答:
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
这个问题已经解决