我怎么知道 tflite_convert 中的 --output_arrays答案

【问题标题】：How do i know my --output_arrays in tflite_convert我怎么知道 tflite_convert 中的 --output_arrays
【发布时间】：2019-11-10 22:07:22
【问题描述】：

我正在尝试使用 tflite_convert 将我的 .pb 转换为 .tflite

我怎么知道我的 --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_

【问题讨论】：

标签： tensorflow object-detection tensorflow-lite

【解决方案1】：

一种选择是使用Netron 来可视化您的图表。

单击特定层会给出其名称，在本例中为：** MobilenetV1/Predictions/Softmax**

希望对你有帮助:)

【讨论】：