MetadataExtractor

MetadataExtractor kelas publik

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

MetadataExtractor (penyangga ByteBuffer )
Membuat MetadataExtractor dengan model TFLite FlatBuffer.

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, seperti 1.14.1 atau 1.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, seperti 1.14.2 .