Как мне узнать свой --output_arrays в tflite_convert

Question

Я пытаюсь преобразовать свой.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

Answer 1

Один из вариантов - использовать Netron для визуализации вашего графика.

Нажатие на конкретный слой дает его имя, которое в данном случае: ** MobilenetV1/Predictions/Softmax**

Я надеюсь, что это помогает:)