如何用我自己的 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 。