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
oder1.14.0
, wirdtrue
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
, wirdfalse
zurückgegeben.