Как я узнаю мои --output_arrays в tflite_convert - PullRequest
Новая
       39

Как я узнаю мои --output_arrays в tflite_convert

0 голосов
/ 11 ноября 2019

Я пытаюсь преобразовать мой .pb в .tflite, используя tflite_convert

Как узнать мои --output_arrays?

Я использую ssd_mobilenet_v2_coco_2018_03_29

Это мой текущий код: 

tflite_convert --output_file=C:/tensorflow1/models/research/object_detection/inference_graph/detect.tflite --graph_def_file=C:/tensorflow1/models/research/object_detection/inference_graph/tflite_graph.pb --inference_type=FLOAT --inference_input_type=QUANTIZED_UINT8 --input_arrays=ImageTensor --input_shapes=1,513,513,3 --output_arrays=SemanticPredictions --mean_values=128 --std_dev_values=128 --allow_custom_ops

, и он выдает ошибку:

Указанный выходной массив«SemanticPredictions» не создается ни одной операцией в этом графе.

Исходя из https://www.tensorflow.org/lite/convert/cmdline_examples#command-line_tools_

...