所有針對 Google 裝置端 AI 高效執行階段的新開發作業,都將專屬於 LiteRT。使用 TensorFlow Lite 套件的應用程式將繼續運作,但所有新更新都只會納入 LiteRT 套件。LiteRT API 包含與 TF Lite API 相同的方法名稱,因此遷移至 LiteRT 不需要進行詳細的程式碼變更。
如需套件名稱變更,請參閱下列特定平台的遷移指南。
使用 Android 遷移
如要遷移使用 TensorFlow Lite 的 Android 應用程式,請將依附元件從 org.tensorflow:tensorflow-lite
改為 com.google.ai.edge.litert
。LiteRT Maven 存放區包含下列套件:
com.google.ai.edge.litert:litert
com.google.ai.edge.litert:litert-gpu
com.google.ai.edge.litert:litert-metadata
com.google.ai.edge.litert:litert-support
您可以在 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。