lớp công khai
Lớp trình bao bọc cho mô hình TFLite và trình thông dịch TFLite.
Lưu ý: Một Model chỉ có thể chứa 1 mô hình TFLite tại một thời điểm và luôn giữ một phiên bản trình thông dịch TFLite để chạy mô hình đó.
Các lớp lồng nhau
| lớp học | Model.Builder | Lớp này không được dùng nữa. Vui lòng sử dụng Model.createModel(Context, String, Options) . | |
| liệt kê | Model.Thiết bị | Loại thiết bị thời gian chạy được sử dụng để thực hiện phân loại. | |
| lớp học | Model.Options | Các tùy chọn để chạy mô hình. | |
Phương pháp công khai
| trống rỗng | đóng () |
| mô hình tĩnh | createModel (Ngữ cảnh ngữ cảnh, tùy chọn String modelPath, Model.Options ) Tải mô hình từ nội dung và khởi tạo trình thông dịch TFLite với các tùy chọn nhất định. |
| mô hình tĩnh | createModel (Ngữ cảnh, String modelPath) Tải mô hình từ nội dung và khởi tạo trình thông dịch TFLite. |
| Bộ đệm MappedByte | lấy dữ liệu () Trả về dữ liệu mô hình được ánh xạ bộ nhớ. |
| Tenxơ | getInputTensor (int inputIndex) Lấy Tensor liên kết với chỉ mục đầu vào được cung cấp. |
| Tenxơ | getOutputTensor (int đầu ra Index) Lấy Tensor liên kết với chỉ số đầu ra được cung cấp. |
| int[] | getOutputTensorShape (int đầu raIndex) Trả về hình dạng đầu ra. |
| Sợi dây | getPath () Trả về đường dẫn của tệp mô hình được lưu trữ trong Nội dung. |
| trống rỗng |
Phương pháp kế thừa
Phương pháp công khai
đóng khoảng trống công khai ()
public static Model createModel (Bối cảnh bối cảnh, String modelPath, Model.Options tùy chọn)
Tải mô hình từ nội dung và khởi tạo trình thông dịch TFLite với các tùy chọn nhất định.
Thông số
| bối cảnh | Bối cảnh ứng dụng. |
|---|---|
| đường dẫn mô hình | Đường dẫn của tệp mô hình. |
| tùy chọn | Các tùy chọn để chạy mô hình. |
Ném
| IOException | nếu có ngoại lệ xảy ra khi mở tệp mô hình. |
|---|
Xem thêm
public static Model createModel (Ngữ cảnh, String modelPath)
Tải mô hình từ nội dung và khởi tạo trình thông dịch TFLite.
Các tùy chọn mặc định là: (1) Thiết bị CPU; (2) một chủ đề.
Thông số
| bối cảnh | Bối cảnh ứng dụng. |
|---|---|
| đường dẫn mô hình | Đường dẫn của tệp mô hình. |
Ném
| IOException | nếu có ngoại lệ xảy ra khi mở tệp mô hình. |
|---|
Tensor công khai getInputTensor (int inputIndex)
Lấy Tensor liên kết với chỉ mục đầu vào được cung cấp.
Thông số
| chỉ số đầu vào |
|---|
Ném
| Ngoại lệ nhà nước bất hợp pháp | nếu trình thông dịch bị đóng. |
|---|
Tensor công khai getOutputTensor (int inputIndex)
Lấy Tensor liên kết với chỉ số đầu ra được cung cấp.
Thông số
| chỉ số đầu ra |
|---|
Ném
| Ngoại lệ nhà nước bất hợp pháp | nếu trình thông dịch bị đóng. |
|---|
int công khai[] getOutputTensorShape (int inputIndex)
Trả về hình dạng đầu ra. Hữu ích nếu hình dạng đầu ra chỉ được xác định khi biểu đồ được tạo.
Thông số
| chỉ số đầu ra |
|---|
Ném
| Ngoại lệ nhà nước bất hợp pháp | nếu trình thông dịch bị đóng. |
|---|
public void run (đầu vào Object[] , đầu ra Map < Integer , Object >)
Chạy suy luận mô hình trên nhiều đầu vào và trả về nhiều đầu ra.
Thông số
| đầu vào | một mảng dữ liệu đầu vào. Các đầu vào phải theo cùng thứ tự với các đầu vào của mô hình. Mỗi đầu vào có thể là một mảng hoặc mảng nhiều chiều hoặc một ByteBuffer thuộc các kiểu nguyên thủy bao gồm int, float, long và byte. ByteBuffer là cách ưa thích để truyền dữ liệu đầu vào lớn, trong khi các loại chuỗi yêu cầu sử dụng đường dẫn đầu vào mảng (đa chiều). Khi ByteBuffer được sử dụng, nội dung của nó sẽ không thay đổi cho đến khi thực hiện suy luận mô hình. |
|---|---|
| kết quả đầu ra | ánh xạ các chỉ mục đầu ra tới các mảng đa chiều của dữ liệu đầu ra hoặc ByteBuffer thuộc các kiểu nguyên thủy bao gồm int, float, long và byte. Nó chỉ cần giữ các mục để đầu ra được sử dụng. |