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
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, come1.14.1
o1.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, come1.14.2
.