Migrar do TensorFlow Lite para o LiteRT

O TensorFlow Lite agora é o LiteRT. O LiteRT é o ambiente de execução principal da IA de alto desempenho no dispositivo do Google. Embora os pacotes atuais do TensorFlow Lite continuem funcionando, todas as atualizações de recursos e melhorias de desempenho futuras serão exclusivas do LiteRT. Como o LiteRT oferece suporte total à API TensorFlow Lite Interpreter, a migração exige apenas uma atualização do nome do pacote. Não é necessário mudar a lógica.

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

Migrar no Android

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

Você pode fazer essa mudança nas dependências do build.gradle:

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

Google Play Services

O LiteRT no tempo de execução do Google Play Services continua usando a dependência play-services-tflite. Se o app já estiver usando o ambiente de execução dos Serviços do Google Play com o TensorFlow Lite, não será necessário fazer mudanças no código.

Para usar o LiteRT nos Serviços do Google Play, adicione o seguinte às dependências do 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 por ai-edge-litert.

Instale a LiteRT com 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 Task Library e a biblioteca Model Maker continuam existindo nos pacotes do TensorFlow Lite. Aplicativos que usam esses SDKs ou bibliotecas não devem migrar para o LiteRT.