将 tensorflow keras LSTM 模型转换为 .tflite 或任何工作格式答案

【问题标题】：Converting tensorflow keras LSTM model into .tflite or any working format将 tensorflow keras LSTM 模型转换为 .tflite 或任何工作格式
【发布时间】：2019-09-27 18:16:32
【问题描述】：

我无法将我的模型转换为可以通过 android-studio 部署在 android 应用程序上的模型。我的模型解释如下：

model = Sequential()
model.add(LSTM(70, activation='relu', input_shape=(n_steps, n_features)))
model.add(Dense(1))
model.compile(optimizer='adam', loss='mse')

然后通过.save函数保存：

model.save("my.h5")

如何在安卓应用上打开这个模型？我已经看到 tflite 在 LSTM 方面存在问题，并且我尝试使用 dl4j 加载它失败了。

【问题讨论】：

您是否尝试使用 TensorFlow Mobile（而不是 TF Lite）加载 .pb 模型？它支持所有操作
没有。我已经看到了一些 yt 比较，但我无法找到任何关于移动设备或添加依赖项的方式的文档。
你可以关注这个blog。

标签： java android tensorflow keras dl4j

【解决方案1】：

使用 tensorflow contrib(lite.TFLiteConverter.from_keras_model_file( 'model.h5')) 中的 Tfite 转换器将 HDF5 转换为 Tflite 模型。

【讨论】：