如何将重新训练的模型转换为tflite格式？

Question

我在MobileNet上重新训练了一个图像分类器模型，我有这些文件。

​

此外，我使用toco压缩重新训练的模型，以将模型转换为.lite格式，但我需要它为.tflite格式。有没有什么方法可以从现有的文件中转换成tflite格式？

Answer 1

下面是一个简单的python脚本，您可以使用它将.pb格式的图形转换为tflite。

import tensorflow as tf

graph_def_file = "output_graph.pb"  ##Your frozen graph

input_arrays = ["input"]         ##Input Node
output_arrays = ["final_result"] ##Output Node

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)

Answer 2

您可以将.lite模型重命名为.tflite，它应该可以正常工作。或者，使用toco，您可以在创建输出时对其进行重命名：

toco \
  --input_file=tf_files/retrained_graph.pb \
  --output_file=tf_files/optimized_graph.lite \ //change this to tflite
  --input_format=TENSORFLOW_GRAPHDEF \
  --output_format=TFLITE \
  --input_shape=1,224,224,3 \
  --input_array=input \
  --output_array=final_result \
  --inference_type=FLOAT \
  --input_data_type=FLOAT