将model.pb转换为model.tflite

时间:2020-02-20 12:08:20

标签: python linux keras model tensorflow-lite

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

一个文件内的整个模式

model.save("model.h5")

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

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

2 个答案:

答案 0 :(得分:0)

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

答案 1 :(得分:0)

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

在TF2.x中,要使用文件转换模型,请使用:

let myInput = document.createElement('input');
myInput.id = 'second';
myInput.type = 'number';
myInput.min = '0';
myInput.step = 'any';
myInput.inputmode = 'decimal';

document.getElementById('here').appendChild(myInput);

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

在TF1.x中,使用以下命令转换文件:

first input: <input id='first' type='number' min='0' step='any' inputmode='decimal'>

<div id='here'>second input: </div>

此处有更多详细信息:https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/g3doc/r1/convert/python_api.md