Migrar do TensorFlow Lite para o LiteRT

Todo novo desenvolvimento do ambiente de execução de alto desempenho do Google para IA no dispositivo vai ser exclusivamente no LiteRT. Os aplicativos que usam pacotes do TensorFlow Lite continuam funcionando, mas todas as novas atualizações só serão incluídas no LiteRT pacotes. As APIs LiteRT contêm os mesmos nomes de métodos que as APIs do TF Lite. Portanto, a migração para a LiteRT não requer alterações detalhadas de código.

Para mudanças no nome dos pacotes, consulte os guias de migração a seguir plataformas.

Migrar com o Android

Para migrar um aplicativo Android usando o Tensorflow Lite, substitua a dependência de org.tensorflow:tensorflow-lite a com.google.ai.edge.litert. A Repositório Maven do LiteRT inclui os seguintes pacotes:

É possível fazer essa mudança nas dependências build.gradle:

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

Google Play Services

O LiteRT no ambiente de execução do Google Play Services continua usando o play-services-tflite . Se o app já estiver usando o ambiente de execução do Google Play Services com o TensorFlow Lite, não é necessário fazer mudanças no código.

Para usar o LiteRT no Google Play Services, adicione o seguinte ao seu build.gradle dependências:

dependencies {
...
    // LiteRT dependencies for Google Play services
    implementation 'com.google.android.gms:play-services-tflite-java:16.0.1'
    // Optional: include LiteRT Support Library
    implementation 'com.google.android.gms:play-services-tflite-support:16.0.1'
...
}

Migrar com Python

Para migrar o código Python usando o Tensorflow Lite, substitua o pacote PIP de tflite-runtime para ai-edge-litert.

Instale o LiteRT com PIP:

$ python3 -m pip install ai-edge-litert

Importe o LiteRT com o seguinte:

from ai_edge_lite.interpreter import Interpreter
interpreter = Interpreter(model_path=args.model_file)

Outras bibliotecas

Os SDKs do Swift e Objective-C para iOS, SDK C++, biblioteca de tarefas e o Model Maker continua existindo nas APIs do TensorFlow Lite. Os aplicativos que usam SDKs ou bibliotecas não podem migrar para o LiteRT.