MetadataExtractor

klasa publiczna MetadataExtractor

Wczytuje metadane z obiektu FlatBuffer dla modelu TFLite.

Obiekt FlatBuffer dla modelu TFLite można wygenerować za pomocą pliku schematu modelu TFLite.

Niektóre modele zawierają bufor płaskiej metadanych TFLite, który rejestruje więcej informacji o tym, co robi model i jak go interpretować. TFLite Metadata Flatbuffer można wygenerować za pomocą pliku schematu metadanych TFLite.

Dozwolone jest przekazywanie modelu FlatBuffer bez metadanych TFLite. Wywoływanie metod, które odczytują metadane z metadanych TFLite, powoduje jednak błędy działania.

Podobnie jest dozwolone przekazywanie do modelu FlatBuffer bez powiązanych plików. Wywoływanie metod, które odczytują powiązane pliki, powoduje jednak błędy podczas działania.

Chociaż model TFLite FlatBuffer obsługuje wiele podtytułów, na razie tłumacz TFLite obsługuje tylko 1 podtytuł. Więcej informacji znajdziesz w instrukcji określania podpunktu podczas konwersji. Z tego powodu funkcja MetadataExtractor pomija indeks podrzędny jako dane wejściowe w swoich metodach.

Zagnieżdżone klasy

klasa MetadataExtractor.QuantizationParams Parametry kwantyzacji odpowiadające tabeli QuantizationParameters w pliku schematu modelu TTite .

Konstruktorzy publiczni

MetadataExtractor(bufor ByteBuffer)
Tworzy MetadataExtractor z użyciem modelu TFLite FlatBuffer.

Metody publiczne

InputStream
getAssociatedFile(String fileName)
Pobiera spakowany powiązany plik z określonym atrybutem fileName.
Set<ciąg znaków>
getAssociatedFileNames()
Pobiera nazwy powiązanych plików.
int
getInputTensorCount(),
Pobiera liczbę tensorów wejściowych w modelu.
TensorMetadata
getInputTensorMetadata(int inputIndex)
Pobiera metadane dla tensora wejściowego określonego przez funkcję inputIndex.
MetadataExtractor.QuantizationParams
getInputTensorQuantizationParams(int inputIndex)
Pobiera parametry kwantyzacji dla tensora wejściowego określonego przez funkcję inputIndex.
int[]
getInputTensorShape(intinputIndex)
Zwraca kształt tensora wejściowego za pomocą funkcji inputIndex.
bajt
getInputTensorType(int inputIndex)
Pobiera ERROR(/TensorType) tensora wejściowego za pomocą funkcji inputIndex.
ModelMetadata
getModelMetadata(),
Pobiera główny moduł obsługi metadanych modelu.
int
getOutputTensorCount()
Pobiera liczbę tensorów wyjściowych w modelu.
TensorMetadata
getOutputTensorMetadata(intoutputIndex)
Pobiera metadane dla tensora wyjściowego określonego przez funkcję outputIndex.
MetadataExtractor.QuantizationParams
getOutputTensorQuantizationParams(intoutputIndex)
Pobiera parametry kwantyzacji dla tensora wyjściowego określonego przez funkcję outputIndex.
int[]
getOutputTensorShape(intoutputIndex)
Zwraca kształt tensora wyjściowego za pomocą funkcji outputIndex.
bajt
getOutputTensorType(intoutputIndex)
Pobiera ERROR(/TensorType) tensora wyjściowego za pomocą funkcji outputIndex.
boolean
hasMetadata(),
Zwraca wartość true, jeśli model ma metadane.
finalna wartość logiczna
isMinimumParserVersionSatisfied()
Zwraca true, jeśli minimalna wersja parsera wymagana przez podany płaski bufor metadanych poprzedza wersję parsera metadanych, z której korzysta ta biblioteka MetadataExtractor, lub jest równa tej wersji.

Metody dziedziczone

Konstruktorzy publiczni

public MetadataExtractor (bufor ByteBuffer)

Tworzy MetadataExtractor z użyciem modelu TFLite FlatBuffer.

Parametry
bufor model TFLite FlatBuffer
Rzuty
IllegalArgumentException jeśli liczba tensorów wejściowych lub wyjściowych w modelu nie odpowiada tej w metadanych
IOException jeśli podczas odczytu modelu jako pliku ZIP wystąpi błąd,

Metody publiczne

public InputStream getAssociatedFile (ciąg znaków fileName)

Pobiera spakowany powiązany plik z określonym atrybutem fileName.

Parametry
fileName nazwa powiązanego pliku
Zwroty
  • nieprzetworzony strumień danych wejściowych zawierający określony plik
Rzuty
IllegalStateException jeśli model nie jest plikiem ZIP
IllegalArgumentException jeśli określony plik nie istnieje w modelu

public Set<String> getAssociatedFileNames ()

Pobiera nazwy powiązanych plików.

Zwroty
  • nazwy powiązanych plików
Rzuty
IllegalStateException jeśli model nie jest plikiem ZIP

public int getInputTensorCount ()

Pobiera liczbę tensorów wejściowych w modelu.

public TensorMetadata getInputTensorMetadata (int inputIndex)

Pobiera metadane dla tensora wejściowego określonego przez funkcję inputIndex.

Parametry
inputIndex indeks żądanego tensora wejściowego
Rzuty
IllegalStateException jeśli ten model nie zawiera metadanych modelu

public MetadataExtractor.QuantizationParams getInputTensorQuantizationParams (int inputIndex)

Pobiera parametry kwantyzacji dla tensora wejściowego określonego przez funkcję inputIndex.

Parametry
inputIndex indeks żądanego tensora wejściowego

public int[] getInputTensorShape (intinputIndex)

Zwraca kształt tensora wejściowego za pomocą funkcji inputIndex.

Parametry
inputIndex indeks żądanego tensora wejściowego

public byte getInputTensorType (int inputIndex)

Pobiera ERROR(/TensorType) tensora wejściowego za pomocą funkcji inputIndex.

Parametry
inputIndex indeks żądanego tensora wejściowego

public ModelMetadata getModelMetadata ()

Pobiera główny moduł obsługi metadanych modelu.

Rzuty
IllegalStateException jeśli ten model nie zawiera metadanych modelu

public int getOutputTensorCount ()

Pobiera liczbę tensorów wyjściowych w modelu.

public TensorMetadata getOutputTensorMetadata (intoutputIndex)

Pobiera metadane dla tensora wyjściowego określonego przez funkcję outputIndex.

Parametry
outputIndex indeks żądanego tensora wyjściowego
Rzuty
IllegalStateException jeśli ten model nie zawiera metadanych modelu

public MetadataExtractor.QuantizationParams getOutputTensorQuantizationParams (intoutputIndex)

Pobiera parametry kwantyzacji dla tensora wyjściowego określonego przez funkcję outputIndex.

Parametry
outputIndex indeks żądanego tensora wyjściowego

public int[] getOutputTensorShape (intoutputIndex)

Zwraca kształt tensora wyjściowego za pomocą funkcji outputIndex.

Parametry
outputIndex indeks żądanego tensora wyjściowego

public byte getOutputTensorType (intoutputIndex)

Pobiera ERROR(/TensorType) tensora wyjściowego za pomocą funkcji outputIndex.

Parametry
outputIndex indeks żądanego tensora wyjściowego

public boolean hasMetadata ()

Zwraca wartość true, jeśli model ma metadane. W przeciwnym razie zwraca false.

public final boolean isMinimumParserVersionSatisfied ()

Zwraca true, jeśli minimalna wersja parsera wymagana przez podany płaski bufor metadanych poprzedza wersję parsera metadanych, z której korzysta ta biblioteka MetadataExtractor, lub jest równa tej wersji. W tym przypadku wszystkie pola metadanych mogą być prawidłowo analizowane za pomocą tej biblioteki wyodrębniania metadanych. W przeciwnym razie zwraca wartość false.

Załóżmy na przykład, że podstawowa wersja parsera metadanych to 1.14.1,

  • zwraca true, jeśli wymagana minimalna wersja parsera jest taka sama lub starsza, na przykład 1.14.1 lub 1.14.0. Wszystkie wersje liczbowe poprzedza wersję null, ponieważ niektóre płaskie bufory metadanych są generowane przed pierwszym wydaniem z różnymi wersjami;
  • zwraca false, jeśli wymagana minimalna wersja parsera jest nowsza, np. 1.14.2.