Migrar do TensorFlow Lite para o LiteRT

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

Para mudanças no nome do pacote, consulte os guias de migração a seguir para plataformas específicas.

Migrar com o Android

Para migrar um app Android usando o TensorFlow Lite, substitua a dependência de org.tensorflow:tensorflow-lite por com.google.ai.edge.litert. O repositório do Maven do LiteRT inclui os seguintes pacotes:

Você pode 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 Play Services continua usando a dependência play-services-tflite. Se o app já estiver usando o ambiente de execução do Play Services com o TensorFlow Lite, não será necessário fazer mudanças no código.

Para usar o LiteRT no Play Services, adicione o seguinte às dependências 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'
...
}

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 o PIP:

$ python3 -m pip install ai-edge-litert

Importe o LiteRT com o seguinte:

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

Outras bibliotecas

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