如何使用经过训练的 keras 模型在 Android 应用程序中进行分类

Question

我正在构建一个 android 应用程序，该应用程序从手机的摄像头获取一些实时图像并将它们分类为不同的类。为此，我首先在 keras 中构建了一个深度学习模型，并且已经使用足够的数据对其进行了训练，并将权重保存在 .hdf5 文件中。现在，我如何使用这个模型对图像进行实时预测，在 android 应用程序中使用 android 相机拍摄。

基本上，有什么方法可以使用 .hdf5 文件中的权重在 Android 应用程序中实时进行预测？
或者，有没有其他方法可以训练模型，以便可以将模型部署在 android 中以在 android 应用程序中进行图像分类。

Answer 1

我是 keras 的新手。但即使我试图通过在 keras 中训练模型并开发一个 android 应用程序来构建一个类似的应用程序。 我看到了 Siraj Raval 的 youtube 视频。 https://www.youtube.com/watch?v=kFWKdLOxykE&t=673s

您需要将权重保存在 .pb 文件中并将模型导出到 android。

请检查是否有任何帮助。