从 TensorFlow Lite 迁移到 LiteRT

TensorFlow Lite 现在是 LiteRT。 LiteRT 是 Google 高性能设备端 AI 的主要运行时。 虽然现有的 TensorFlow Lite 软件包仍可正常运行，但未来的所有功能更新和性能增强都将仅适用于 LiteRT。由于 LiteRT 完全支持 TensorFlow Lite Interpreter API，因此迁移只需更新软件包名称，无需更改任何逻辑。

如需了解软件包名称更改，请参阅以下针对特定平台的迁移指南。

在 Android 上迁移

如需迁移使用 TensorFlow Lite 的 Android 应用，请将依赖项从 org.tensorflow:tensorflow-lite 替换为 com.google.ai.edge.litert。LiteRT Maven 代码库包含以下软件包：

您可以在 build.gradle 依赖项中进行此项更改：

dependencies {
  ...
  implementation `com.google.ai.edge.litert:litert:2.1.0`
}

Play 服务

Play 服务运行时中的 LiteRT 继续使用 play-services-tflite 依赖项。如果您的应用已将 Play 服务运行时与 TensorFlow Lite 搭配使用，则无需进行任何代码更改。

如需在 Play 服务中使用 LiteRT，请将以下内容添加到 build.gradle 依赖项中：

dependencies {
...
    // LiteRT dependencies for Google Play services
    implementation 'com.google.android.gms:play-services-tflite-java:16.4.0'
    // Optional: include LiteRT Support Library
    implementation 'com.google.android.gms:play-services-tflite-support:16.4.0'
...
}

使用 Python 进行迁移

如需迁移使用 TensorFlow Lite 的 Python 代码，请将 PIP 软件包从 tflite-runtime 替换为 ai-edge-litert。

使用 PIP 安装 LiteRT：

$ python3 -m pip install ai-edge-litert

使用以下代码导入 LiteRT：

from ai_edge_litert.interpreter import Interpreter
interpreter = Interpreter(model_path=args.model_file)

其他库

适用于 iOS 的 Swift 和 Objective-C SDK、C++ SDK、Task 库和 Model Maker 库仍存在于 TensorFlow Lite 软件包中。使用这些 SDK 或库的应用不应迁移到 LiteRT。