TensorFlow Lite 使用 TensorFlow 模型,这些模型已转换为更小、更高效的机器学习 (ML) 模型格式。您可以使用 TensorFlow Lite 的预训练模型,修改现有模型,或构建自己的 TensorFlow 模型,然后将其转换为 TensorFlow Lite 格式。TensorFlow Lite 模型可以执行常规 TensorFlow 模型几乎可以执行的任何任务:对象检测、自然语言处理、模式识别等等,使用各种输入数据,包括图像、视频、音频和文本。

跳到 转换 部分,了解有关让您的模型与 TensorFlow Lite 一起运行的信息。
有关获取适合您用例的模型的指南,请 继续阅读

您无需构建 TensorFlow Lite 模型即可开始在移动设备或边缘设备上使用机器学习。许多已构建和优化的模型可供您立即在您的应用程序中使用。您可以从在 TensorFlow Lite 中使用预训练模型开始,并随着时间的推移逐步构建自定义模型,如下所示

  1. 使用已 训练的模型 开始开发机器学习功能。
  2. 使用 模型制作器 等工具修改现有的 TensorFlow Lite 模型。
  3. 使用 TensorFlow 工具构建 自定义模型,然后将其 转换为 TensorFlow Lite。

如果您想快速使用机器学习实现功能或实用任务,在使用 TensorFlow Lite 开始开发之前,您应该查看 ML Kit 支持的用例。此开发工具提供您可以直接从移动应用程序调用的 API,以完成常见的 ML 任务,例如条形码扫描和设备内翻译。使用这种方法可以帮助您快速获得结果。但是,ML Kit 在扩展其功能方面选项有限。有关更多信息,请参阅 ML Kit 开发人员文档。


如果为您的特定用例构建自定义模型是您的最终目标,您应该从开发和训练 TensorFlow 模型或扩展现有模型开始。在开始模型开发过程之前,您应该了解 TensorFlow Lite 模型的约束,并在构建模型时牢记这些约束。

  • 有限的计算能力
  • 模型的大小
  • 数据的大小
  • 支持的 TensorFlow 操作

有关这些约束的更多详细信息,请参阅模型构建概述中的 模型设计约束。有关为 TensorFlow Lite 构建有效、兼容、高性能模型的更多信息,请参阅 性能最佳实践

了解如何选择预训练的 ML 模型与 TensorFlow Lite 一起使用。
使用 TensorFlow Lite Model Maker 使用您的训练数据修改模型。
了解如何构建自定义 TensorFlow 模型以与 TensorFlow Lite 一起使用。