我正在构建一个 android 应用程序,它从手机摄像头获取一些实时图像并将它们分类为不同的类。为此,我首先在 keras 中构建了一个深度学习模型,并且已经使用足够的数据对其进行了训练,并将权重保存在 .hdf5 文件中。现在,我如何使用此模型对在 Android 应用程序中使用 Android 相机拍摄的图像进行实时预测。
基本上,有什么方法可以使用 .hdf5 文件中的权重在 Android 应用程序中进行实时预测?
或者,是否有任何其他方法可以训练模型,以便可以将模型部署在 android 中以在 android 应用程序中进行图像分类。
最佳答案
我是keras的新手。但即使我正在尝试通过在 keras 中训练模型并开发 android 应用程序来构建类似的应用程序。 我看到了 Siraj Raval 的 youtube 视频。 https://www.youtube.com/watch?v=kFWKdLOxykE&t=673s
您需要将权重保存在.pb 文件中并将模型导出到android。
请检查是否有帮助。
关于android - 如何使用经过训练的 keras 模型在 Android 应用程序中进行分类,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44669578/