Di chuyển từ TensorFlow Lite sang LiteRT

Tất cả quy trình phát triển mới cho môi trường thời gian chạy hiệu suất cao của Google dành cho AI trên thiết bị sẽ chỉ có trên LiteRT. Ứng dụng dùng gói TensorFlow Lite sẽ tiếp tục hoạt động nhưng tất cả bản cập nhật mới sẽ chỉ có trong LiteRT . API LiteRT chứa cùng tên phương thức với API TF Lite, vì vậy Việc di chuyển sang LiteRT không yêu cầu thay đổi mã chi tiết.

Để thay đổi tên gói, hãy xem các hướng dẫn di chuyển sau đây để biết nền tảng.

Di chuyển bằng Android

Để di chuyển một ứng dụng Android bằng Tensorflow Lite, hãy thay thế phần phụ thuộc từ org.tensorflow:tensorflow-lite đến com.google.ai.edge.litert. Chiến lược phát hành đĩa đơn Kho lưu trữ Maven LiteRT bao gồm các gói sau:

Bạn có thể thực hiện thay đổi này trong các phần phụ thuộc build.gradle:

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

Dịch vụ Play

LiteRT trong thời gian chạy Dịch vụ Play tiếp tục sử dụng play-services-tflite phần phụ thuộc. Nếu ứng dụng của bạn đang sử dụng môi trường thời gian chạy Dịch vụ Play với TensorFlow Lite, bạn không cần thay đổi mã.

Để sử dụng LiteRT trong Dịch vụ Play, hãy thêm thông tin sau vào build.gradle của bạn phần phụ thuộc:

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'
...
}

Di chuyển bằng Python

Để di chuyển mã Python bằng Tensorflow Lite, hãy thay thế gói PIP từ tflite-runtime thành ai-edge-litert.

Cài đặt LiteRT có PIP:

$ python3 -m pip install ai-edge-litert

Nhập LiteRT với lệnh sau:

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

Thư viện khác

SDK Swift và Mục tiêu-C dành cho iOS, SDK C++, Thư viện tác vụ và Trình tạo mô hình thư viện vẫn tiếp tục tồn tại trong API TensorFlow Lite. Các ứng dụng dùng Không nên di chuyển SDK hoặc thư viện sang LiteRT.