保存/传输模型 - Android 上的 TensorFlow Lite 迁移学习答案

【问题标题】：Saving/Transmitting Model - TensorFlow Lite Transfer Learning on Android保存/传输模型 - Android 上的 TensorFlow Lite 迁移学习
【发布时间】：2020-12-07 03:37:19
【问题描述】：

我正在尝试创建一对 Android 应用：一个用于训练图像分类迁移学习模型，另一个仅使用经过训练的模型进行推理。这些应用程序将在不同的设备上运行，其用途在于在功能更强大的设备上训练模型，并能够在功能更弱的可穿戴设备上使用该模型进行推理。迁移学习正在实施，如以下帖子所述：https://blog.tensorflow.org/2019/12/example-on-device-model-personalization.html。

问题是我找不到保存训练模型并将其从第一个设备传输到第二个设备的好方法。我尝试过实现蓝牙传输的序列化，但是Android TFL库不容易序列化。在 Android 上以某种方式保存 .tflite 文件有多难？此功能是否已经存在而我错过了？任何帮助或想法将不胜感激。谢谢！

【问题讨论】：

您好，您找到保存模型的好方法了吗？

标签： android tensorflow serialization android-bluetooth tensorflow-lite

【解决方案1】：

为了传输模型，您应该将其作为二进制文件执行，而不是尝试显式序列化/反序列化。 Android 上有许多不同的库可用于此，因此找到适合您的应用的东西应该不会太难。

至于加载 TFLite 模型本身并运行推理，可以使用 TFLite 解释器类在设备本地执行此操作，并将其简单地指向您的设备上文件。你可以在这里找到一个例子：https://www.tensorflow.org/lite/inference_with_metadata/lite_support

【讨论】：