從 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.0.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。