在android studio中查找两个类之间的简单图像分类代码的问题

Question

我在 2 个类上训练了 keras 模型并将其转换为 .tflite 模型。现在，当我从图库中放置图像时，我想在 android studio 中使用此模型在两个类之间进行简单分类。关于这种简单的方法，我在互联网上找不到任何帮助。在互联网上有相机的方法，但我在我的简单项目中不需要。

Answer 1

您需要使用 TensorFlow Lite 的 Java API 在设备上运行推理。从Android quickstart开始，如果你还没有安装依赖等。

如果您的模型与标准图像分类模型（例如 MobileNet）的工作方式相似，您可以从 TFLite Image Classification app 来源获取灵感。具体来说，您可能对Classifier base-class（及其floating point 子类）感兴趣。分类器演示了如何使用 TFLite 的 Java API 来实例化新的解释器并运行图像输入推理。