将 model.pb 转换为 model.tflite

Question

我有一个使用 keras 构建的 ML 翻译模式，我有 50 个 checkoint_epoch.hdf5 然后我找到了一种保存方法

一个文件中的整个模式是

model.save("model.h5")

然后我找到了一个工具，我将此文件转换为model.pb现在我必须将其更改为model.tflite以便在ionic v4 和 angular 的移动应用程序中使用它

以及如何将model.pb转换为model.tflite

Answer 1

参见https://www.tensorflow.org/lite/convert有 python API 和命令行工具。

Answer 2

最简单的方法是使用 python API。

在 TF2.x 中，要从文件转换模型，请使用：

converter = tf.lite.TFLiteConverter.from_saved_model(export_dir)
tflite_model = converter.convert()

可以在此处找到更多详细信息： https : //www.tensorflow.org/lite/convert/python_api

在 TF1.x 中，使用以下命令转换文件：

graph_def_file = "/path/model.pb"
input_arrays = ["input"]
# Change the following to your application
output_arrays = ["MobilenetV1/Predictions/Softmax"]

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)

更多细节在这里： https : //github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/g3doc/r1/convert/python_api.md