如何用我自己的 tflite 替换 Tensorflow 的 Android 图像分类示例中的 tflite？

Question

我克隆了Tensorflow 的 Android 图像分类示例，现在我想使用从这里下载的我自己的 tflite 对食物进行分类。 如何用我自己的 tflite 替换默认的 tflite？ 我只有一个 tflite，而在 Android 示例中有一对，所以我不确定要摆脱哪个 tflite。

Answer 1

首先，将 model 添加到assets文件夹中。 或者，您可以直接从 TFHub 下载 model，方法与download.gradle 中的模型类似。

那么如果你使用任务库来运行推理（更简单），你需要添加一个类似的ClassifierFloatEfficientNet.java的 class 并提供 Z20F35E630DAF44DBFA4C3F68F539D 信息。 在Classifier.java中注册您的新 class 。

如果您使用 TFLite Support 库来运行推理，您将像这样添加新的 class 。