如何在 .tflite 模型上执行迁移学习

Question

目前，我正在尝试制作一个Android食物识别应用程序。我设法构建一个应用程序来处理图像并在“mobilenet_quant_v1_224.tflite”预训练模型上运行。一切正常，现在我想将自己的调用添加到模型中进行预测。

我确实知道TFLITE还处于初级阶段，也许有些人能够突破障碍。如果是这样，请帮忙。任何回复/建议/方法都会有帮助，因为我不知道如何继续。

即使没有任何方法可以解决它，请建议我一种方法来实现此应用程序的目标。

链接到经过训练的模型：http://download.tensorflow.org/models/mobilenet_v1_2018_02_22/mobilenet_v1_1.0_224_quant.tgz

真心希望能得到一些帮助。

再次提前致谢。

Answer 1

截至目前，TFLite 不支持训练。您可以对 TF 模型进行迁移学习，然后将迁移学习模型转换为 TFLite。此TF-for-poets-2-tflite Codelab将准确引导您完成该过程（包括在 TF 模型上进行迁移学习的链接）。