從 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.litertLiteRT 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、工作程式庫和 Model Maker 程式庫,仍會保留在 TensorFlow Lite 套件中。使用這些 SDK 或程式庫的應用程式不應遷移至 LiteRT。