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")

如何在 Android 应用程序上打开此模型？我发现 tflite 在 LSTM 方面存在问题，并且我尝试使用 dl4j 加载它失败了。

Answer 1

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