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

Question

我无法将我的模型转换为可以通过 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 加载它失败了。

Answer 1

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