Keras深度学习模型到android

Vaj*_*aka 9 android deep-learning keras tensorflow

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

Ale*_*lex 8

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

  • 将Tensflow配置为Keras后端.

  • 保存模型使用model.save(filepath)(你已经完成了)

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

解决方案1:在Tensflow中导入模型

1-构建Tensorflow模型

2-构建Android应用程序并调用tensflow.查看本教程和谷歌的官方演示,了解如何操作.

解决方案2:在java中导入模型
1- deeplearning4j java库允许导入keras模型:教程链接
2-在Android中使用deeplearning4j:由于你处于java世界中,因此很容易.查看本教程