Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand

使用 Python 快速入门基于 Linux 的设备

对于基于 Linux 的嵌入式设备（例如 Raspberry Pi 和使用 Edge TPU 的 Coral 设备），非常适合将 TensorFlow Lite 与 Python 结合使用。

本页介绍如何在几分钟内学会开始使用 Python 运行 TensorFlow Lite 模型。您只需要一个已转换为 TensorFlow Lite 的 TensorFlow 模型。（如果还没有转换的模型，您可以使用随下面链接的示例提供的模型进行实验。）

关于 TensorFlow Lite 运行时软件包

为了快速开始使用 Python 执行 TensorFlow Lite 模型，您可以仅安装 TensorFlow Lite 解释器，无需安装所有 TensorFlow 软件包。我们将这种简化的 Python 软件包称为 tflite_runtime。

tflite_runtime 软件包是整个 tensorflow 软件包的一小部分，并且包括使用 TensorFlow Lite 运行推断所需的最少代码（主要是 Interpreter Python 类）。当您只想执行 .tflite 模型并避免因使用大型 TensorFlow 库而浪费磁盘空间时，这种小型软件包是理想选择。

注：如果您需要访问其他 Python API（如 TensorFlow Lite Converter），则必须安装完整的 TensorFlow 软件包。例如，tflite_runtime 软件包中不包括 Select TF 算子。如果您的模型与 Select TF 算子有任何依赖关系，则需要改用完整的 TensorFlow 软件包。

安装适用于 Python 的 TensorFlow Lite

您可以使用 pip 在 Linux 上安装：

python3 -m pip install tflite-runtime

受支持的平台

tflite-runtime Python wheel 针对以下平台进行预构建并提供给这些平台：

  • Linux armv7l（例如运行 32 位 Raspberry Pi OS 的 Raspberry Pi 2、3、4 和 Zero 2）
  • Linux aarch64（例如运行 Debian ARM64 的 Raspberry Pi 3、4）
  • Linux x86_64

如果您想在其他平台上运行 TensorFlow Lite 模型，则应使用完整 TensorFlow 软件包，或者从源代码构建 tflite-runtime 软件包。

如果您将 TensorFlow 与 Coral Edge TPU 结合使用，则应遵循相应的 Coral 设置文档。

注：我们已不再更新 Debian 软件包 python3-tflite-runtime。最新版 Debian 软件包适用于 TF 2.5 版本，您可以按照这些早期说明进行安装。

注：我们已不再发布适用于 Windows 和 macOS 的预构建 tflite-runtime wheel。对于这些平台，您应使用完整 TensorFlow 软件包，或者从源代码构建 tflite-runtime 软件包。

使用 tflite_runtime 运行推断

现在，您需要从 tflite_runtime 导入 Interpreter，而不是从 tensorflow 模块导入。

例如，安装上述软件包后，如果复制并运行 label_image.py 文件，（可能）会失败，因为您没有安装 tensorflow 库。要解决此问题，请编辑该文件中的下面一行：

import tensorflow as tf

将其改成：

import tflite_runtime.interpreter as tflite

然后更改下面一行：

interpreter = tf.lite.Interpreter(model_path=args.model_file)

将其改成：

interpreter = tflite.Interpreter(model_path=args.model_file)

现在，重新运行 label_image.py。就是这样！您现在执行的正是 TensorFlow Lite 模型。

了解详情