lớp công khai
Mô hình
Lớp trình bao bọc cho mô hình TFLite và trình phiên dịch TFLite.
Lưu ý: Mỗi lần, Model chỉ có thể lưu 1 kiểu máy TFLite và luôn lưu giữ một phiên bản phiên dịch TFLite để chạy phiên bản đó.
Lớp lồng ghép
| lớp | Model.Builder |
Lớp này không còn được dùng nữa. Vui lòng sử dụng Model.createModel(Context, String, Options).
|
|
| enum | Model.Device | Loại thiết bị thời gian chạy được dùng để thực thi hoạt động phân loại. | |
| lớp | Model.Options | Các tuỳ chọn chạy mô hình. | |
Phương thức công khai
| void |
Hàm close()
|
| tĩnh Mô hình |
createModel(Ngữ cảnh theo bối cảnh, Chuỗi modelPath, tuỳ chọn Model.Options)
Tải một mô hình từ các thành phần và khởi chạy trình phiên dịch TFLite với các tuỳ chọn cho sẵn.
|
| tĩnh Mô hình |
createModel(Ngữ cảnh theo ngữ cảnh, String modelPath)
Tải một mô hình từ các thành phần và khởi chạy trình phiên dịch TFLite.
|
| MappedByteBuffer |
getData()
Trả về dữ liệu mô hình được liên kết với bộ nhớ.
|
| Tensor |
getInputTensor(int inputIndex)
Lấy Tensor liên kết với chỉ mục đầu vào được cung cấp.
|
| Tensor |
getOutputTensor(int outputIndex)
Lấy Tensor liên kết với chỉ mục đầu ra được cung cấp.
|
| int[] |
getOutputTensorShape(int outputIndex)
Trả về hình dạng đầu ra.
|
| String |
Hàm getPath()
Trả về đường dẫn của tệp mô hình được lưu trữ trong mục Thành phần.
|
| void |
Các phương thức kế thừa
Phương thức công khai
công khai void close ()
công khai tĩnh Mô hình createModel (Ngữ cảnh ngữ cảnh, String modelPath, tuỳ chọn Model.Options)
Tải một mô hình từ các thành phần và khởi chạy trình phiên dịch TFLite với các tuỳ chọn cho sẵn.
Thông số
| bối cảnh | Ngữ cảnh ứng dụng. |
|---|---|
| modelPath | Đường dẫn của tệp mô hình. |
| tùy chọn | Các tuỳ chọn để chạy mô hình. |
Gửi
| IOException | nếu có bất kỳ ngoại lệ nào xảy ra khi mở tệp mô hình. |
|---|
Xem thêm
công khai tĩnh Mô hình createModel (Ngữ cảnh ngữ cảnh, String modelPath)
Tải một mô hình từ các thành phần và khởi chạy trình phiên dịch TFLite.
Các tuỳ chọn mặc định là: (1) thiết bị CPU; (2) một luồng.
Thông số
| bối cảnh | Ngữ cảnh ứng dụng. |
|---|---|
| modelPath | Đường dẫn của tệp mô hình. |
Gửi
| IOException | nếu có bất kỳ ngoại lệ nào xảy ra khi mở tệp mô hình. |
|---|
công khai Tensor 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ố
| inputIndex |
|---|
Gửi
| IllegalStateException | khi phiên dịch đã đóng. |
|---|
công khai Tensor getOutputTensor (int outputIndex)
Lấy Tensor liên kết với chỉ mục đầu ra được cung cấp.
Thông số
| outputIndex |
|---|
Gửi
| IllegalStateException | khi phiên dịch đã đóng. |
|---|
công khai int[] getOutputTensorShape (int outputIndex)
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 tạo biểu đồ.
Thông số
| outputIndex |
|---|
Gửi
| IllegalStateException | khi phiên dịch đã đóng. |
|---|
công khai void run (Object[] đầu vào, Map<Số nguyên, Kết quả đầu ra Đối tượng>)
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ố
| giá trị đầu vào | một mảng dữ liệu đầu vào. Các dữ liệu đầu vào phải có cùng thứ tự với dữ liệu đầu vào của mô hình. Mỗi dữ liệu đầu vào có thể là một mảng hoặc mảng đa chiều hoặc ByteBuffer gồm các loại nguyên gốc bao gồm int, float, long và byte. ByteBuffer là cách ưu tiên để truyền dữ liệu đầu vào có kích thước lớn, trong khi các loại chuỗi yêu cầu sử dụng đường dẫn đầu vào của 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 suy luận về mô hình. |
|---|---|
| kết quả đầu ra | ánh xạ các chỉ mục đầu ra ánh xạ đến các mảng đa chiều của dữ liệu đầu ra hoặc ByteBuffer thuộc các loại nguyên gốc bao gồm int, float, long và byte. Hàm này chỉ cần giữ lại các mục nhập để sử dụng kết quả.
|