Tải siêu dữ liệu từ TFLite Model FlatBuffer.
Bộ đệm phẳng mô hình TFLite có thể được tạo bằng tệp lược đồ Mô hình TFLite.
Một số mô hình chứa Bộ đệm phẳng siêu dữ liệu TFLite, ghi lại nhiều thông tin hơn về chức năng của mô hình và cách diễn giải mô hình. Bộ đệm phẳng siêu dữ liệu TFLite có thể được tạo bằng tệp lược đồ siêu dữ liệu TFLite.
Nó được phép chuyển vào mô hình FlatBuffer mà không có siêu dữ liệu TFLite. Tuy nhiên, việc gọi các phương thức đọc từ siêu dữ liệu TFLite sẽ gây ra lỗi thời gian chạy.
Tương tự, nó được phép chuyển vào FlatBuffer mô hình mà không cần các tệp liên quan. Tuy nhiên, việc gọi các phương thức đọc các tệp liên quan sẽ gây ra lỗi thời gian chạy.
Mặc dù mô hình TFLite FlatBuffer hỗ trợ nhiều đồ thị con, nhưng cho đến nay, Trình thông dịch TFLite chỉ hỗ trợ một đồ thị con duy nhất. Xem hướng dẫn về cách chỉ định đồ thị con trong quá trình chuyển đổi để biết thêm thông tin. Do đó, MetadataExtractor
bỏ qua chỉ mục đồ thị con làm đầu vào trong các phương thức của nó.
Các lớp lồng nhau
lớp học | MetadataExtractor.QuantizationParams | Các tham số lượng tử hóa tương ứng với bảng QuantizationParameters trong tệp lược đồ Mô hình TFLite. |
Nhà xây dựng công cộng
Phương pháp công khai
Luồng đầu vào | |
Đặt < Chuỗi > | getAssociatedFileNames () Lấy tên tập tin của các tập tin liên quan. |
int | getInputTensorCount () Lấy số lượng tensor đầu vào trong mô hình. |
Siêu dữ liệu Tensor | getInputTensorMetadata (int inputIndex) Lấy siêu dữ liệu cho tensor đầu vào được chỉ định bởi inputIndex . |
MetadataExtractor.QuantizationParams | getInputTensorQuantizationParams (int inputIndex) Lấy các tham số lượng tử hóa cho tenxơ đầu vào được chỉ định bởi inputIndex . |
int[] | getInputTensorShape (int inputIndex) Nhận hình dạng của tensor đầu vào với inputIndex . |
byte | getInputTensorType (int inputIndex) Nhận ERROR(/TensorType) của tensor đầu vào với inputIndex . |
Siêu dữ liệu mô hình | getModelMetadata () Nhận trình xử lý gốc cho siêu dữ liệu mô hình. |
int | getOutputTensorCount () Lấy số lượng tensor đầu ra trong mô hình. |
Siêu dữ liệu Tensor | getOutputTensorMetadata (int đầu ra Index) Lấy siêu dữ liệu cho tenxơ đầu ra được chỉ định bởi outputIndex . |
MetadataExtractor.QuantizationParams | getOutputTensorQuantizationParams (int inputIndex) Lấy các tham số lượng tử hóa cho tenxơ đầu ra được chỉ định bởi outputIndex . |
int[] | getOutputTensorShape (int đầu ra Index) Nhận hình dạng của tenxơ đầu ra với outputIndex . |
byte | getOutputTensorType (int đầu ra Index) Lấy ERROR(/TensorType) của tenxơ đầu ra với outputIndex . |
boolean | hasMetadata () Trả về true nếu mô hình có siêu dữ liệu. |
boolean cuối cùng | isMinimumParserVersionSatisfied () Trả về true nếu phiên bản trình phân tích cú pháp tối thiểu được yêu cầu bởi bộ đệm siêu dữ liệu đã cho có trước hoặc bằng với phiên bản của trình phân tích cú pháp siêu dữ liệu mà thư viện MetadataExtractor này đang dựa vào. |
Phương pháp kế thừa
Nhà xây dựng công cộng
MetadataExtractor công khai (bộ đệm ByteBuffer )
Tạo một MetadataExtractor
với FlatBuffer mô hình TFLite.
Thông số
đệm | mô hình FlatBuffer của TFLite |
---|
Ném
Ngoại lệ Đối số bất hợp pháp | nếu số lượng tensor đầu vào hoặc đầu ra trong mô hình không khớp với số lượng trong siêu dữ liệu |
---|---|
IOException | nếu xảy ra lỗi khi đọc mô hình dưới dạng tệp Zip |
Phương pháp công khai
public inputStream getAssociatedFile ( Tên tệp chuỗi)
Lấy tệp liên kết được đóng gói với fileName
được chỉ định.
Thông số
tên tập tin | tên của tập tin liên quan |
---|
Trả lại
- luồng đầu vào thô chứa tệp được chỉ định
Ném
Ngoại lệ nhà nước bất hợp pháp | nếu mô hình không phải là tệp zip |
---|---|
Ngoại lệ Đối số bất hợp pháp | nếu tệp được chỉ định không tồn tại trong mô hình |
Bộ công khai < Chuỗi > getAssociatedFileNames ()
Lấy tên tập tin của các tập tin liên quan.
Trả lại
- tên tập tin của các tập tin liên quan
Ném
Ngoại lệ nhà nước bất hợp pháp | nếu mô hình không phải là tệp zip |
---|
int công khai getInputTensorCount ()
Lấy số lượng tensor đầu vào trong mô hình.
công khai TensorMetadata getInputTensorMetadata (int inputIndex)
Lấy siêu dữ liệu cho tensor đầu vào được chỉ định bởi inputIndex
.
Thông số
chỉ số đầu vào | chỉ số của tensor đầu vào mong muốn |
---|
Ném
Ngoại lệ nhà nước bất hợp pháp | nếu mô hình này không chứa siêu dữ liệu mô hình |
---|
công khai MetadataExtractor.QuantizationParams getInputTensorQuantizationParams (int inputIndex)
Lấy các tham số lượng tử hóa cho tenxơ đầu vào được chỉ định bởi inputIndex
.
Thông số
chỉ số đầu vào | chỉ số của tensor đầu vào mong muốn |
---|
int công khai[] getInputTensorShape (int inputIndex)
Nhận hình dạng của tensor đầu vào với inputIndex
.
Thông số
chỉ số đầu vào | chỉ số của tensor đầu vào mong muốn |
---|
byte công khai getInputTensorType (int inputIndex)
Nhận ERROR(/TensorType)
của tensor đầu vào với inputIndex
.
Thông số
chỉ số đầu vào | chỉ số của tensor đầu vào mong muốn |
---|
công khai ModelMetadata getModelMetadata ()
Nhận trình xử lý gốc cho siêu dữ liệu mô hình.
Ném
Ngoại lệ nhà nước bất hợp pháp | nếu mô hình này không chứa siêu dữ liệu mô hình |
---|
int công khai getOutputTensorCount ()
Lấy số lượng tensor đầu ra trong mô hình.
công khai TensorMetadata getOutputTensorMetadata (int inputIndex)
Lấy siêu dữ liệu cho tenxơ đầu ra được chỉ định bởi outputIndex
.
Thông số
chỉ số đầu ra | chỉ số của tensor đầu ra mong muốn |
---|
Ném
Ngoại lệ nhà nước bất hợp pháp | nếu mô hình này không chứa siêu dữ liệu mô hình |
---|
công khai MetadataExtractor.QuantizationParams getOutputTensorQuantizationParams (int inputIndex)
Lấy các tham số lượng tử hóa cho tenxơ đầu ra được chỉ định bởi outputIndex
.
Thông số
chỉ số đầu ra | chỉ số của tensor đầu ra mong muốn |
---|
int công khai[] getOutputTensorShape (int inputIndex)
Nhận hình dạng của tenxơ đầu ra với outputIndex
.
Thông số
chỉ số đầu ra | chỉ số của tensor đầu ra mong muốn |
---|
byte công khai getOutputTensorType (int inputIndex)
Lấy ERROR(/TensorType)
của tenxơ đầu ra với outputIndex
.
Thông số
chỉ số đầu ra | chỉ số của tensor đầu ra mong muốn |
---|
boolean công khai hasMetadata ()
Trả về true
nếu mô hình có siêu dữ liệu. Ngược lại, trả về false
.
boolean cuối cùng công khai isMinimumParserVersionSatisfied ()
Trả về true
nếu phiên bản trình phân tích cú pháp tối thiểu được yêu cầu bởi bộ đệm siêu dữ liệu đã cho có trước hoặc bằng với phiên bản của trình phân tích cú pháp siêu dữ liệu mà thư viện MetadataExtractor này đang dựa vào. Tất cả các trường trong siêu dữ liệu có thể được phân tích cú pháp chính xác bằng thư viện trình trích xuất siêu dữ liệu này trong trường hợp này. Nếu không, nó sẽ trả về false
.
Ví dụ: giả sử phiên bản trình phân tích cú pháp siêu dữ liệu cơ bản là 1.14.1
,
- nó trả về
true
, nếu phiên bản trình phân tích cú pháp tối thiểu được yêu cầu giống hoặc cũ hơn, chẳng hạn như1.14.1
hoặc1.14.0
. Phiên bản Null đi trước tất cả các phiên bản số, vì một số bộ đệm phẳng siêu dữ liệu được tạo trước bản phát hành có phiên bản đầu tiên; - nó trả về
false
nếu phiên bản trình phân tích cú pháp tối thiểu được yêu cầu mới hơn, chẳng hạn như1.14.2
.