MetadataExtractor

classe pubblica MetadataExtractor

Carica i metadati dal modello TFLite FlatBuffer.

Il modello TFLite FlatBuffer può essere generato utilizzando il file di schema del modello TFLite.

Alcuni modelli contengono un Flatbuffer dei metadati TFLite, che registra ulteriori informazioni su cosa fa il modello e come interpretarlo. Il Flatbuffer dei metadati TFLite può essere generato utilizzando il file di schema dei metadati TFLite.

È consentito passare in un modello FlatBuffer senza metadati TFLite. Tuttavia, il richiamo di metodi che leggono dai metadati TFLite causerà errori di runtime.

Allo stesso modo, è consentito passare in un modello FlatBuffer senza file associati. Tuttavia, il richiamo di metodi che leggono i file associati causerà errori di runtime.

Sebbene il modello TFLite FlatBuffer supporti più sottografi, TFLite Interpreter finora supporta solo un singolo sottografo. Per ulteriori informazioni, vedere le istruzioni su come specificare il sottografo durante la conversione. Pertanto, MetadataExtractor omette l'indice del sottografo come input nei suoi metodi.

Classi nidificate

classe MetadataExtractor.QuantizationParams Parametri di quantizzazione che corrispondono alla tabella QuantizationParameters nel file di schema del modello TFLite.

Costruttori pubblici

MetadataExtractor (buffer ByteBuffer )
Crea un MetadataExtractor con il modello TFLite FlatBuffer.

Metodi pubblici

InputStream
getAssociatedFile ( String nomefile)
Ottiene il file associato compresso con il fileName specificato.
Impostare < Stringa >
getNomiFileAssociati ()
Ottiene i nomi dei file associati.
int
getInputTensorCount ()
Ottiene il conteggio dei tensori di input nel modello.
TensorMetadata
getInputTensorMetadata (int inputIndex)
Ottiene i metadati per il tensore di input specificato da inputIndex .
MetadataExtractor.QuantizationParams
getInputTensorQuantizationParams (int inputIndex)
Ottiene i parametri di quantizzazione per il tensore di input specificato da inputIndex .
int[]
getInputTensorShape (int inputIndex)
Ottiene la forma del tensore di input con inputIndex .
byte
getInputTensorType (int inputIndex)
Ottiene ERROR(/TensorType) del tensore di input con inputIndex .
Metadati del modello
getModelMetadata ()
Ottiene il gestore root per i metadati del modello.
int
getOutputTensorCount ()
Ottiene il conteggio dei tensori di output nel modello.
TensorMetadata
getOutputTensorMetadata (int outputIndex)
Ottiene i metadati per il tensore di output specificato da outputIndex .
MetadataExtractor.QuantizationParams
getOutputTensorQuantizationParams (int outputIndex)
Ottiene i parametri di quantizzazione per il tensore di output specificato da outputIndex .
int[]
getOutputTensorShape (int outputIndex)
Ottiene la forma del tensore di output con outputIndex .
byte
getOutputTensorType (int outputIndex)
Ottiene ERROR(/TensorType) del tensore di output con outputIndex .
booleano
hasMetadata ()
Restituisce true se il modello ha metadati.
booleano finale
isMinimumParserVersionSatisfied ()
Restituisce true se la versione minima del parser richiesta dal flatbuffer dei metadati specificato precede o è uguale alla versione del parser dei metadati su cui fa affidamento questa libreria MetadataExtractor.

Metodi ereditati

Costruttori pubblici

public MetadataExtractor (buffer ByteBuffer )

Crea un MetadataExtractor con il modello TFLite FlatBuffer.

Parametri
respingente il modello TFLite FlatBuffer
Lancia
IllegalArgumentException se il numero di tensori di input o output nel modello non corrisponde a quello nei metadati
IOException se si verifica un errore durante la lettura del modello come file Zip

Metodi pubblici

public InputStream getAssociatedFile ( String fileName)

Ottiene il file associato compresso con il fileName specificato.

Parametri
nome del file il nome del file associato
ritorna
  • il flusso di input non elaborato contenente il file specificato
Lancia
IllegalStateException se il modello non è un file zip
IllegalArgumentException se il file specificato non esiste nel modello

public Set < String > getAssociatedFileNames ()

Ottiene i nomi dei file associati.

ritorna
  • i nomi dei file associati
Lancia
IllegalStateException se il modello non è un file zip

public int getInputTensorCount ()

Ottiene il conteggio dei tensori di input nel modello.

public TensorMetadata getInputTensorMetadata (int inputIndex)

Ottiene i metadati per il tensore di input specificato da inputIndex .

Parametri
inputIndex l'indice del tensore di ingresso desiderato
Lancia
IllegalStateException se questo modello non contiene metadati del modello

public MetadataExtractor.QuantizationParams getInputTensorQuantizationParams (int inputIndex)

Ottiene i parametri di quantizzazione per il tensore di input specificato da inputIndex .

Parametri
inputIndex l'indice del tensore di ingresso desiderato

public int[] getInputTensorShape (int inputIndex)

Ottiene la forma del tensore di input con inputIndex .

Parametri
inputIndex l'indice del tensore di ingresso desiderato

byte pubblico getInputTensorType (int inputIndex)

Ottiene ERROR(/TensorType) del tensore di input con inputIndex .

Parametri
inputIndex l'indice del tensore di ingresso desiderato

public ModelMetadata getModelMetadata ()

Ottiene il gestore root per i metadati del modello.

Lancia
IllegalStateException se questo modello non contiene metadati del modello

public int getOutputTensorCount ()

Ottiene il conteggio dei tensori di output nel modello.

public TensorMetadata getOutputTensorMetadata (int outputIndex)

Ottiene i metadati per il tensore di output specificato da outputIndex .

Parametri
outputIndice l'indice del tensore di uscita desiderato
Lancia
IllegalStateException se questo modello non contiene metadati del modello

public MetadataExtractor.QuantizationParams getOutputTensorQuantizationParams (int outputIndex)

Ottiene i parametri di quantizzazione per il tensore di output specificato da outputIndex .

Parametri
outputIndice l'indice del tensore di uscita desiderato

public int[] getOutputTensorShape (int outputIndex)

Ottiene la forma del tensore di output con outputIndex .

Parametri
outputIndice l'indice del tensore di uscita desiderato

byte pubblico getOutputTensorType (int outputIndex)

Ottiene ERROR(/TensorType) del tensore di output con outputIndex .

Parametri
outputIndice l'indice del tensore di uscita desiderato

hasMetadata booleano pubblico ()

Restituisce true se il modello ha metadati. Altrimenti restituisce false .

booleano finale pubblico isMinimumParserVersionSatisfied ()

Restituisce true se la versione minima del parser richiesta dal flatbuffer dei metadati specificato precede o è uguale alla versione del parser dei metadati su cui fa affidamento questa libreria MetadataExtractor. In questo caso, tutti i campi nei metadati possono essere analizzati correttamente con questa libreria di estrazione dei metadati. Altrimenti restituisce false .

Ad esempio, supponiamo che la versione del parser dei metadati sottostante sia 1.14.1 ,

  • restituisce true , se la versione minima del parser richiesta è la stessa o precedente, come 1.14.1 o 1.14.0 . La versione nulla precede tutte le versioni numeriche, poiché alcuni flatbuffer di metadati vengono generati prima del primo rilascio della versione;
  • restituisce false , se la versione minima del parser richiesta è più recente, come 1.14.2 .