如何使用自定义Tensorflow Lite模型

Question

我对机器学习非常陌生。我有一个带有非常简单的TensorFlow模型的python文件，我需要使用Google的ML Kit在Android上进行部署（这将创建tflite文件）。我绝对不明白python文件的结构应该是什么，Google的文档也没有使它变得更容易。也许有人有一个很好的例子，可以从SCRATCH转换编写的CUSTOM模型，然后在Java中使用它。我需要从Android的文本字段传递一个字符串，并获得预期的答案。

Answer 1

您首先需要在任何数据集上训练模型。模型中的图层必须符合TFLite库支持的图层。这是list个受支持和不受支持的图层。

一旦您对它进行了培训，将根据您的保存方式（例如，使用kerasmodel.save）。遵循此tutorial或此页上的其他教程，将其转换为TFLite。

现在您可以在Android Studio中使用此.tflite模型。为此，您可以遵循良好的tutorial。