TensorFlow Lite - 自定義 model 被替換為示例 model

Question

我正在通過為 Android 構建二進制圖像分類應用程序來學習 TensorFlow lite。 I trained my tflite model by following the official model maker tutorial ( https://www.tensorflow.org/lite/tutorials/model_maker_image_classification ), where I got my model.tflite and label.txt file saved in the google colab directory. As per the TensorFlow lite tutorial ( https://www.tensorflow.org/lite/models/image_classification/overview ), I added model.tflite and label.txt (containing the labels of my classes) in assets folder on the reference android應用程序。 並更改了代碼中的 label 文件名和參考 android 應用程序的代碼中的 TensorFlow Lite 文件名。

However, while running the reference app in Android studio, I am still seeing the default model provided by TensorFlow using the default classes for classification, and not the model I trained and classes of classification I created.

Answer 1

go to download_model.gradle each time the app is built, the gradle download the cefault model replace it with your model link (upload your model to google drive and put the link in the gradle)