如何将保存的.pb模型转换为TFLITE？

Question

我有一个经过TF 1.4训练的模型，该模型已导出到冻结的推理图中，并带有文件“ models / research / object_detection / export_tflite_ssd_graph.py”

如何将其转换为tflite？我遇到很多问题

Answer 1

您可以使用command line tool或Python API。

Python API示例：

converter = tf.lite.TFLiteConverter.from_frozen_graph(
  graph_def_file, input_arrays, output_arrays)
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)

CLI示例：

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

Tensorflow正式建议使用Python API。