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



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

我确实知道TFLITE仍处于初步阶段,也许有些人设法打破了障碍。如果是这样,请帮忙。任何回复/建议/方法都会有所帮助,因为我不知道如何进行。

即使没有任何方法可以绕过它,请向我建议一种实现此应用程序目标的方法。

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

真的希望得到一些帮助。

再次提前感谢。

截至目前,TFLite 不支持培训。 您可以在 TF 模型上进行迁移学习,然后将迁移学习模型转换为 TFLite。 这个 TF-for-poets-2-tflite Codelab 将引导您完成此操作(包括在您的 TF 模型上进行迁移学习的链接(。

最新更新