将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模型。