Граф tflight создан неправильно - PullRequest
Новая
       20

Граф tflight создан неправильно

0 голосов
/ 23 апреля 2019

У меня есть следующий замороженный вывод логики. Это для семантической сегментации с использованием 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

После конвертации график выглядит следующим образом ( скачать его здесь ) Мой вопрос: как мне получить график, аналогичный доступному графику googles для Deeplab ( здесь )? Чтобы дать вам более ясный вопрос, см. Изображение ниже enter image description here

График слева - это мой график tflite, а график справа - это график deeplab от Google. Как получить результаты, похожие на график справа?

1 Ответ

0 голосов
/ 30 апреля 2019 
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

Это решило мой вопрос

...