android - 如何在 .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 模型上进行迁移学习的链接)。