python - 将Tensorflow模型转换为Tensorflow Lite

我将张量流模型转换为tflite有问题。
我有一个基于Tensorflow Object Detection的学习模型
我想使用TFlite converter中的转换代码

curl https://storage.googleapis.com/download.tensorflow.org/models/mobilenet_v1_0.50_128_frozen.tgz | tar xzv -C /tmp

tflite_convert \
 --output_file=/tmp/foo.tflite \
 --graph_def_file=/tmp/mobilenet_v1_0.50_128/frozen_graph.pb \
 --input_arrays=input \
 --output_arrays=MobilenetV1/Predictions/Reshape_1

我不知道从哪里获得input_arrays和output_arrays的值。

感谢您的回答

最佳答案

我们在对象检测API中有a script来获取Flatbuffer。