android - Keras深度学习模型到android

Question

我正在为 Android 开发一个实时对象分类应用程序。首先，我使用“keras”创建了一个深度学习模型，并且我已经将经过训练的模型保存为“model.h5”文件。我想知道如何在 android 中使用该模型进行图像分类。

Answer 1

您不能将 Keras 直接导出到 Android，但您必须保存模型

• 将 Tensorflow 配置为您的 Keras 后端。

• 使用 model.save(filepath) 保存模型权重(您已经这样做了)

然后使用以下解决方案之一加载它:

方案一:在Tensflow中导入模型

1- 构建 Tensorflow 模型

• 从 keras 模型构建 tensorflow 模型 use this code (链接已更新)

2- 构建 Android 应用程序并调用 Tensorflow。检查这个tutorial还有这个official demo从谷歌学习如何去做。

方案二:在java中导入模型
1- deeplearning4j 一个 java 库允许导入 keras 模型:tutorial link
2- 在 Android 中使用 deeplearning4j:这很容易，因为你在 java 世界中。检查this tutorial