Memuat metadata dari TFLite Model FlatBuffer.
TFLite Model FlatBuffer dapat dibuat menggunakan file skema TFLite Model.
Beberapa model berisi TFLite Metadata Flatbuffer, yang mencatat lebih banyak informasi tentang fungsi model dan cara menafsirkan model. Metadata TFLite Flatbuffer dapat dibuat menggunakan file skema Metadata TFLite.
Diperbolehkan meneruskan model FlatBuffer tanpa metadata TFLite. Namun, memanggil metode yang membaca dari metadata TFLite akan menyebabkan kesalahan runtime.
Demikian pula, diperbolehkan untuk meneruskan model FlatBuffer tanpa file terkait. Namun, memanggil metode yang membaca file terkait akan menyebabkan kesalahan runtime.
Meskipun model TFLite FlatBuffer mendukung banyak subgraf, sejauh ini TFLite Interpreter hanya mendukung satu subgraf. Lihat instruksi cara menentukan subgraf selama konversi untuk informasi lebih lanjut. Oleh karena itu, MetadataExtractor
menghilangkan indeks subgraf sebagai masukan dalam metodenya.
Kelas Bersarang
kelas | MetadataExtractor.QuantizationParams | Parameter kuantisasi yang sesuai dengan tabel, QuantizationParameters , dalam file skema Model TFLite. |
Konstruktor Publik
Metode Publik
Aliran Masukan | getAssociatedFile ( String nama file) Mendapatkan file terkait yang dikemas dengan fileName yang ditentukan. |
Setel < Tali > | dapatkan Nama File Terkait () Mendapatkan nama file dari file terkait. |
ke dalam | dapatkanInputTensorCount () Mendapatkan jumlah tensor masukan dalam model. |
Metadata Tensor | getInputTensorMetadata (int inputIndex) Mendapatkan metadata untuk tensor masukan yang ditentukan oleh inputIndex . |
MetadataExtractor.QuantizationParams | getInputTensorQuantizationParams (int inputIndex) Mendapatkan parameter kuantisasi untuk tensor masukan yang ditentukan oleh inputIndex . |
ke dalam[] | getInputTensorShape (int inputIndex) Mendapatkan bentuk tensor masukan dengan inputIndex . |
byte | getInputTensorType (int inputIndex) Mendapatkan ERROR(/TensorType) dari tensor input dengan inputIndex . |
ModelMetadata | dapatkanModelMetadata () Mendapatkan pengendali root untuk metadata model. |
ke dalam | dapatkanOutputTensorCount () Mendapatkan jumlah tensor keluaran dalam model. |
Metadata Tensor | getOutputTensorMetadata (int keluaranIndeks) Mendapatkan metadata untuk tensor keluaran yang ditentukan oleh outputIndex . |
MetadataExtractor.QuantizationParams | getOutputTensorQuantizationParams (int outputIndex) Mendapatkan parameter kuantisasi untuk tensor keluaran yang ditentukan oleh outputIndex . |
ke dalam[] | getOutputTensorShape (int keluaranIndeks) Mendapatkan bentuk tensor keluaran dengan outputIndex . |
byte | getOutputTensorType (int keluaranIndeks) Mendapatkan ERROR(/TensorType) dari tensor keluaran dengan outputIndex . |
boolean | memiliki Metadata () Mengembalikan true jika model memiliki metadata. |
boolean terakhir | isMinimumParserVersionSatisfied () Mengembalikan true jika versi parser minimum yang diperlukan oleh flatbuffer metadata tertentu mendahului atau sama dengan versi parser metadata yang diandalkan oleh pustaka MetadataExtractor ini. |
Metode Warisan
Konstruktor Publik
MetadataExtractor publik (penyangga ByteBuffer )
Membuat MetadataExtractor
dengan model TFLite FlatBuffer.
Parameter
penyangga | model TFLite FlatBuffer |
---|
Melempar
Pengecualian Argumen Ilegal | jika jumlah tensor input atau output dalam model tidak sesuai dengan metadata |
---|---|
Pengecualian IO | jika terjadi kesalahan saat membaca model sebagai file Zip |
Metode Publik
InputStream publik getAssociatedFile ( String nama file)
Mendapatkan file terkait yang dikemas dengan fileName
yang ditentukan.
Parameter
nama file | nama file terkait |
---|
Kembali
- aliran input mentah yang berisi file tertentu
Melempar
IlegalStateException | jika modelnya bukan file zip |
---|---|
Pengecualian Argumen Ilegal | jika file yang ditentukan tidak ada dalam model |
Set publik < String > getAssociatedFileNames ()
Mendapatkan nama file dari file terkait.
Kembali
- nama file dari file terkait
Melempar
IlegalStateException | jika modelnya bukan file zip |
---|
int publik getInputTensorCount ()
Mendapatkan jumlah tensor masukan dalam model.
TensorMetadata publik getInputTensorMetadata (int inputIndex)
Mendapatkan metadata untuk tensor masukan yang ditentukan oleh inputIndex
.
Parameter
masukanIndeks | indeks tensor masukan yang diinginkan |
---|
Melempar
IlegalStateException | jika model ini tidak berisi metadata model |
---|
MetadataExtractor.QuantizationParams publik getInputTensorQuantizationParams (int inputIndex)
Mendapatkan parameter kuantisasi untuk tensor masukan yang ditentukan oleh inputIndex
.
Parameter
masukanIndeks | indeks tensor masukan yang diinginkan |
---|
int publik[] getInputTensorShape (int inputIndex)
Mendapatkan bentuk tensor masukan dengan inputIndex
.
Parameter
masukanIndeks | indeks tensor masukan yang diinginkan |
---|
byte publik getInputTensorType (int inputIndex)
Mendapatkan ERROR(/TensorType)
dari tensor input dengan inputIndex
.
Parameter
masukanIndeks | indeks tensor masukan yang diinginkan |
---|
ModelMetadata publik getModelMetadata ()
Mendapatkan pengendali root untuk metadata model.
Melempar
IlegalStateException | jika model ini tidak berisi metadata model |
---|
int publik getOutputTensorCount ()
Mendapatkan jumlah tensor keluaran dalam model.
TensorMetadata publik getOutputTensorMetadata (int outputIndex)
Mendapatkan metadata untuk tensor keluaran yang ditentukan oleh outputIndex
.
Parameter
Indeks keluaran | indeks tensor keluaran yang diinginkan |
---|
Melempar
IlegalStateException | jika model ini tidak berisi metadata model |
---|
MetadataExtractor.QuantizationParams getOutputTensorQuantizationParams (int outputIndex)
Mendapatkan parameter kuantisasi untuk tensor keluaran yang ditentukan oleh outputIndex
.
Parameter
Indeks keluaran | indeks tensor keluaran yang diinginkan |
---|
int publik[] getOutputTensorShape (int outputIndex)
Mendapatkan bentuk tensor keluaran dengan outputIndex
.
Parameter
Indeks keluaran | indeks tensor keluaran yang diinginkan |
---|
byte publik getOutputTensorType (int outputIndex)
Mendapatkan ERROR(/TensorType)
dari tensor keluaran dengan outputIndex
.
Parameter
Indeks keluaran | indeks tensor keluaran yang diinginkan |
---|
boolean publik hasMetadata ()
Mengembalikan true
jika model memiliki metadata. Jika tidak, kembalikan false
.
boolean akhir publik isMinimumParserVersionSatisfied ()
Mengembalikan true
jika versi parser minimum yang diperlukan oleh flatbuffer metadata tertentu mendahului atau sama dengan versi parser metadata yang diandalkan oleh pustaka MetadataExtractor ini. Dalam hal ini, semua bidang dalam metadata dapat diurai dengan benar menggunakan pustaka ekstraktor metadata ini. Jika tidak, ia akan mengembalikan false
.
Misalnya, asumsikan versi parser metadata yang mendasarinya adalah 1.14.1
,
- ia mengembalikan
true
, jika versi parser minimum yang diperlukan sama atau lebih lama, seperti1.14.1
atau1.14.0
. Versi null mendahului semua versi numerik, karena beberapa flatbuffer metadata dihasilkan sebelum rilis versi pertama; - ia mengembalikan
false
, jika versi parser minimum yang diperlukan lebih baru, seperti1.14.2
.