有什么方法可以在 flutter 中部署自定义 ML 模型?

is there any way to deploy custom ML models in flutter?

我正在尝试使用下面的 ML 模型创建一个 flutter 应用 link。

https://www.tensorflow.org/lite/models/style_transfer/overview

您可以将自定义的 tensorwflow 模型上传到 firebase ML KIT(自定义选项卡),并在您的 flutter 项目中与 firebase API 集成。

我没有玩过 Flutter,但有一些很有前途的文章 article 1 article 2 article 3 提供了插入 tensorflow lite 依赖项、创建资产文件夹和加载模型的代码片段。您可以检查它们,我希望您对如何进行有一个初步的了解。不过,风格迁移是一个中型到硬编码项目。我建议从更简单的任务开始,如分类问题。

编码愉快

最近发布了一个用于集成任意 TFLite 模型的 flutter 插件。

https://pub.dev/packages/tflite_flutter

请参阅 pub 包的自述文件,了解如何捆绑 .tflite 模型以及如何在 flutter 中加载/运行 它们。

有一篇博客post介绍如何使用这个flutter插件,但是它使用了不同的模型(文本分类)作为示例。

https://medium.com/@am15hg/text-classification-using-tensorflow-lite-plugin-for-flutter-3b92f6655982

我之前在flutter app中使用过这个style_transfer模型。不幸的是,我丢失了该代码(我没有使用 git)。但我可以就此提出建议。

要使用此自定义模型,您必须使用 https://pub.dev/packages/tflite_flutter

这里的技巧是您必须查看模型的输入形状 (image_shape=(384,384,3), style_shape=(256,256,3) 和输出大小 (shape=(384,384, 3) 以及瓶颈的形状)。您可以使用 https://pub.dev/packages/image.

调整图像大小

有了这两个,你就可以开始了。抱歉英语不好,我知道我来晚了,但也许其他人会发现它有用。

编码愉快!

已编辑: 所以我找到了代码。我将它的副本存储在我的 google 驱动器中。我已将其上传到 github https://github.com/Rizwan2613/style-transfer。请务必阅读自述文件。谢谢