将 model.pb 转换为 model.tflite答案

【问题标题】：convert model.pb to model.tflite将 model.pb 转换为 model.tflite
【发布时间】：2020-02-20 12:08:20
【问题描述】：

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

一个文件内的整个模式

model.save("model.h5")

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

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

【问题讨论】：

标签： python linux keras model tensorflow-lite

【解决方案1】：

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

【讨论】：

【解决方案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)

【讨论】：