MetadataExtractor

öffentliche Klasse MetadataExtractor

Lädt Metadaten aus dem TFLite-Modell-FlatBuffer.

Der TFLite-Modell-FlatBuffer kann mithilfe der TFLite-Modellschemadatei generiert werden.

Einige Modelle enthalten einen TFLite Metadata Flatbuffer, der weitere Informationen darüber aufzeichnet, was das Modell tut und wie es interpretiert wird. Der TFLite-Metadaten-Flatbuffer kann mit der TFLite-Metadatenschemadatei generiert werden.

Es darf ein Modell-FlatBuffer ohne TFLite-Metadaten übergeben werden. Das Aufrufen von Methoden, die aus TFLite-Metadaten lesen, führt jedoch zu Laufzeitfehlern.

Ebenso ist es zulässig, einen Modell-FlatBuffer ohne verknüpfte Dateien zu übergeben. Methoden, die die zugehörigen Dateien lesen, führen jedoch zu Laufzeitfehlern.

Obwohl das TFLite-Modell FlatBuffer mehrere Teilgrafiken unterstützt, unterstützt TFLite Interpreter bisher nur eine einzelne Teilgrafik. Weitere Informationen Daher lässt MetadataExtractor den Untergrafikindex als Eingabe in seinen Methoden weg.

Verschachtelte Klassen

Klasse MetadataExtractor.QuantizationParams Quantisierungsparameter, die der Tabelle QuantizationParameters in der Schemadatei des TFLite-Modells  entsprechen.

Public Constructors

MetadataExtractor(ByteBuffer-Zwischenspeicher)
Erstellt einen MetadataExtractor mit TFLite-Modell-FlatBuffer.

Public Methods

InputStream
getAssociatedFile(String fileName)
Ruft die gepackte verknüpfte Datei mit dem angegebenen fileName ab.
Set<String>
getAssociatedFileNames()
Ruft die Dateinamen der verknüpften Dateien ab.
int
getInputTensorCount()
Ruft die Anzahl der Eingabetensoren im Modell ab.
TensorMetadata
getInputTensorMetadata(int inputIndex)
Ruft die Metadaten für den durch inputIndex angegebenen Eingabetensor ab.
MetadataExtractor.QuantizationParams
getInputTensorQuantizationParams(int inputIndex)
Ruft die Quantisierungsparameter für den durch inputIndex angegebenen Eingabetensor ab.
int[]
getInputTensorShape(int inputIndex)
Ruft die Form des Eingabetensors mit inputIndex ab.
Byte
getInputTensorType(int inputIndex)
Ruft den ERROR(/TensorType) des Eingabetensors mit inputIndex ab.
ModelMetadata
getModelMetadata()
Ruft den Root-Handler für die Modellmetadaten ab.
int
getOutputTensorCount()
Ruft die Anzahl der Ausgabetensoren im Modell ab.
TensorMetadata
getOutputTensorMetadata(int outputIndex)
Ruft die Metadaten für den durch outputIndex angegebenen Ausgabetensor ab.
MetadataExtractor.QuantizationParams
getOutputTensorQuantizationParams(int outputIndex)
Ruft die Quantisierungsparameter für den durch outputIndex angegebenen Ausgabetensor ab.
int[]
getOutputTensorShape(int outputIndex)
Ruft die Form des Ausgabetensors mit outputIndex ab.
Byte
getOutputTensorType(int outputIndex)
Ruft den ERROR(/TensorType) des Ausgabetensors mit outputIndex ab.
boolean
hasMetadata()
Gibt true zurück, wenn das Modell Metadaten hat.
finaler boolescher Wert
isMinimumParserVersionSatisfied()
Gibt true zurück, wenn die für den angegebenen Metadaten-Flatpuffer erforderliche Mindestparserversion der Version des Metadatenparsers voraus ist oder dieser entspricht, auf die sich diese MetadataExtractor-Bibliothek verlässt.

Übernommene Methoden

Public Constructors

public MetadataExtractor (ByteBuffer-Zwischenspeicher)

Erstellt einen MetadataExtractor mit TFLite-Modell-FlatBuffer.

Parameter
Puffer FlatBuffer des TFLite-Modells
Löst aus
IllegalArgumentException Die Anzahl der Eingabe- oder Ausgabetensoren im Modell stimmt nicht mit der Anzahl in den Metadaten überein.
IOException wenn beim Lesen des Modells als ZIP-Datei ein Fehler auftritt.

Public Methods

public InputStream getAssociatedFile (String fileName)

Ruft die gepackte verknüpfte Datei mit dem angegebenen fileName ab.

Parameter
fileName den Namen der verknüpften Datei
Rückgabe
  • den Roheingabestream, der die angegebene Datei enthält
Löst aus
IllegalStateException Wenn das Modell keine ZIP-Datei ist
IllegalArgumentException Die angegebene Datei ist im Modell nicht vorhanden.

public Set<String> getAssociatedFileNames ()

Ruft die Dateinamen der verknüpften Dateien ab.

Rückgabe
  • die Dateinamen der verknüpften Dateien
Löst aus
IllegalStateException Wenn das Modell keine ZIP-Datei ist

public int getInputTensorCount ()

Ruft die Anzahl der Eingabetensoren im Modell ab.

public TensorMetadata getInputTensorMetadata (int inputIndex)

Ruft die Metadaten für den durch inputIndex angegebenen Eingabetensor ab.

Parameter
inputIndex Index des gewünschten Eingabetensors
Löst aus
IllegalStateException Wenn dieses Modell keine Modellmetadaten enthält

public MetadataExtractor.QuantizationParams getInputTensorQuantizationParams (int inputIndex)

Ruft die Quantisierungsparameter für den durch inputIndex angegebenen Eingabetensor ab.

Parameter
inputIndex Index des gewünschten Eingabetensors

public int[] getInputTensorShape (int inputIndex)

Ruft die Form des Eingabetensors mit inputIndex ab.

Parameter
inputIndex Index des gewünschten Eingabetensors

public Byte getInputTensorType (int inputIndex)

Ruft den ERROR(/TensorType) des Eingabetensors mit inputIndex ab.

Parameter
inputIndex Index des gewünschten Eingabetensors

public ModelMetadata getModelMetadata ()

Ruft den Root-Handler für die Modellmetadaten ab.

Löst aus
IllegalStateException Wenn dieses Modell keine Modellmetadaten enthält

public int getOutputTensorCount ()

Ruft die Anzahl der Ausgabetensoren im Modell ab.

public TensorMetadata getOutputTensorMetadata (int outputIndex)

Ruft die Metadaten für den durch outputIndex angegebenen Ausgabetensor ab.

Parameter
outputIndex Index des gewünschten Ausgabetensors
Löst aus
IllegalStateException Wenn dieses Modell keine Modellmetadaten enthält

public MetadataExtractor.QuantizationParams getOutputTensorQuantizationParams (int outputIndex)

Ruft die Quantisierungsparameter für den durch outputIndex angegebenen Ausgabetensor ab.

Parameter
outputIndex Index des gewünschten Ausgabetensors

public int[] getOutputTensorShape (int editionIndex)

Ruft die Form des Ausgabetensors mit outputIndex ab.

Parameter
outputIndex Index des gewünschten Ausgabetensors

public Byte getOutputTensorType (intoutputIndex)

Ruft den ERROR(/TensorType) des Ausgabetensors mit outputIndex ab.

Parameter
outputIndex Index des gewünschten Ausgabetensors

public boolesch hasMetadata ()

Gibt true zurück, wenn das Modell Metadaten hat. Andernfalls wird false zurückgegeben.

public final boolesch isMinimumParserVersionSatisfied ()

Gibt true zurück, wenn die für den angegebenen Metadaten-Flatpuffer erforderliche Mindestparserversion der Version des Metadatenparsers voraus ist oder dieser entspricht, auf die sich diese MetadataExtractor-Bibliothek verlässt. Alle Felder in den Metadaten können in diesem Fall mit dieser Metadaten-Extrahiererbibliothek korrekt geparst werden. Andernfalls wird false zurückgegeben.

Angenommen, die Version des zugrunde liegenden Metadaten-Parsers lautet 1.14.1.

  • Wenn die erforderliche Mindest-Parserversion gleich oder älter ist, z. B. 1.14.1 oder 1.14.0, wird true zurückgegeben. Alle numerischen Versionen haben Vorrang, da einige Metadaten-Flatpuffer vor dem ersten versionierten Release generiert werden.
  • Wenn die erforderliche Mindest-Parserversion neuer ist, z. B. 1.14.2, wird false zurückgegeben.