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 存放區包含下列套件:
com.google.ai.edge.litert:litertcom.google.ai.edge.litert:litert-gpucom.google.ai.edge.litert:litert-metadatacom.google.ai.edge.litert:litert-support
你可以在 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。