我正在通过为Android构建二进制图像分类应用程序来学习TensorFlow lite。我按照官方的模型制造商教程训练了我的tflite模型(https://www.tensorflow.org/lite/tutorials/model_maker_image_classification),在那里我将model.tflite
和label.txt
文件保存在google colab目录中。根据TensorFlow lite教程(https://www.tensorflow.org/lite/models/image_classification/overview),我在参考android应用程序的assets文件夹中添加了model.tflite和label.txt(包含我的类的标签(。并更改了参考android应用程序代码中的标签文件名和TensorFlow Lite文件名。
然而,在Android studio中运行参考应用程序时,我仍然看到TensorFlow提供的默认模型使用默认的分类类,而不是我训练的模型和我创建的分类类。
每次构建应用程序时,都会转到download_model.gradle,gradle下载cefault模型并将其替换为您的模型链接(将您的模型上传到google drive并将链接放在gradle中(