從 TensorFlow Lite 遷移至 LiteRT

Google 裝置端 AI 高效能執行階段的所有新開發項目,都將僅適用於 LiteRT。使用 TensorFlow Lite 套件的應用程式仍可正常運作,但所有新更新只會納入 LiteRT 套件。LiteRT API 包含與 TF Lite API 相同的方法名稱,因此遷移至 LiteRT 時,不需要進行詳細的程式碼變更。

如要變更套件名稱,請參閱下列特定平台的遷移指南。

使用 Android 裝置遷移

如要遷移使用 Tensorflow Lite 的 Android 應用程式,請將依附元件從 org.tensorflow:tensorflow-lite 替換為 com.google.ai.edge.litertLiteRT Maven 存放區包含下列套件:

你可以在 build.gradle 依附元件中進行這項變更:

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

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 API 中。使用這些 SDK 或程式庫的應用程式不應遷移至 LiteRT。