如何在.tflite模型上进行转移学习

Question

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

我知道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模型上进行转移学习的链接）。