Carrega metadados do TFLite Model FlatBuffer.
TFLite Model FlatBuffer pode ser gerado usando o arquivo de esquema do TFLite Model.
Alguns modelos contêm um Flatbuffer de metadados TFLite, que registra mais informações sobre o que o modelo faz e como interpretá-lo. O TFLite Metadata Flatbuffer pode ser gerado usando o arquivo de esquema TFLite Metadata.
É permitido passar um modelo FlatBuffer sem metadados TFLite. No entanto, invocar métodos que leem metadados do TFLite causará erros de tempo de execução.
Da mesma forma, é permitido passar um modelo FlatBuffer sem arquivos associados. No entanto, invocar métodos que leem os arquivos associados causará erros de tempo de execução.
Embora o modelo TFLite FlatBuffer suporte vários subgráficos, o TFLite Interpreter suporta apenas um único subgráfico até o momento. Consulte as instruções sobre como especificar o subgráfico durante a conversão para obter mais informações. Portanto, MetadataExtractor
omite o índice de subgráfico como entrada em seus métodos.
Classes aninhadas
aula | MetadataExtractor.QuantizationParams | Parâmetros de quantização que correspondem à tabela QuantizationParameters , no arquivo de esquema do modelo TFLite. |
Construtores Públicos
Métodos Públicos
Fluxo de entrada | getAssociatedFile ( String nomeArquivo) Obtém o arquivo associado compactado com o fileName especificado. |
Definir < Sequência > | getAssociatedFileNames () Obtém os nomes dos arquivos associados. |
interno | getInputTensorCount () Obtém a contagem de tensores de entrada no modelo. |
TensorMetadados | getInputTensorMetadata (int inputIndex) Obtém os metadados do tensor de entrada especificado por inputIndex . |
MetadataExtractor.QuantizationParams | getInputTensorQuantizationParams (int inputIndex) Obtém os parâmetros de quantização para o tensor de entrada especificado por inputIndex . |
interno[] | getInputTensorShape (int inputIndex) Obtém a forma do tensor de entrada com inputIndex . |
byte | getInputTensorType (int inputIndex) Obtém o ERROR(/TensorType) do tensor de entrada com inputIndex . |
ModeloMetadados | getModelMetadata () Obtém o manipulador raiz dos metadados do modelo. |
interno | getOutputTensorCount () Obtém a contagem de tensores de saída no modelo. |
TensorMetadados | getOutputTensorMetadata (int outputIndex) Obtém os metadados do tensor de saída especificado por outputIndex . |
MetadataExtractor.QuantizationParams | getOutputTensorQuantizationParams (int outputIndex) Obtém os parâmetros de quantização para o tensor de saída especificado por outputIndex . |
interno[] | getOutputTensorShape (int outputIndex) Obtém a forma do tensor de saída com outputIndex . |
byte | getOutputTensorType (int outputIndex) Obtém o ERROR(/TensorType) do tensor de saída com outputIndex . |
boleano | hasMetadados () Retorna true se o modelo tiver metadados. |
booleano final | isMinimumParserVersionSatisfied () Retorna true se a versão mínima do analisador exigida pelo flatbuffer de metadados fornecido precede ou é igual à versão do analisador de metadados em que esta biblioteca MetadataExtractor depende. |
Métodos herdados
Construtores Públicos
público MetadataExtractor (buffer ByteBuffer )
Cria um MetadataExtractor
com modelo TFLite FlatBuffer.
Parâmetros
amortecedor | o modelo TFLite FlatBuffer |
---|
Lança
Exceção de argumento ilegal | se o número de tensores de entrada ou saída no modelo não corresponder ao dos metadados |
---|---|
IOException | se ocorrer um erro ao ler o modelo como um arquivo Zip |
Métodos Públicos
public InputStream getAssociatedFile ( String fileName)
Obtém o arquivo associado compactado com o fileName
especificado.
Parâmetros
nome do arquivo | o nome do arquivo associado |
---|
Devoluções
- o fluxo de entrada bruto contendo o arquivo especificado
Lança
IllegalStateException | se o modelo não for um arquivo zip |
---|---|
Exceção de argumento ilegal | se o arquivo especificado não existir no modelo |
conjunto público <String> getAssociatedFileNames ()
Obtém os nomes dos arquivos associados.
Devoluções
- os nomes dos arquivos associados
Lança
IllegalStateException | se o modelo não for um arquivo zip |
---|
público int getInputTensorCount ()
Obtém a contagem de tensores de entrada no modelo.
público TensorMetadata getInputTensorMetadata (int inputIndex)
Obtém os metadados do tensor de entrada especificado por inputIndex
.
Parâmetros
índice de entrada | o índice do tensor de entrada desejado |
---|
Lança
IllegalStateException | se este modelo não contiver metadados de modelo |
---|
público MetadataExtractor.QuantizationParams getInputTensorQuantizationParams (int inputIndex)
Obtém os parâmetros de quantização para o tensor de entrada especificado por inputIndex
.
Parâmetros
índice de entrada | o índice do tensor de entrada desejado |
---|
public int[] getInputTensorShape (int inputIndex)
Obtém a forma do tensor de entrada com inputIndex
.
Parâmetros
índice de entrada | o índice do tensor de entrada desejado |
---|
byte público getInputTensorType (int inputIndex)
Obtém o ERROR(/TensorType)
do tensor de entrada com inputIndex
.
Parâmetros
índice de entrada | o índice do tensor de entrada desejado |
---|
ModelMetadata pública getModelMetadata ()
Obtém o manipulador raiz dos metadados do modelo.
Lança
IllegalStateException | se este modelo não contiver metadados de modelo |
---|
público int getOutputTensorCount ()
Obtém a contagem de tensores de saída no modelo.
public TensorMetadata getOutputTensorMetadata (int outputIndex)
Obtém os metadados do tensor de saída especificado por outputIndex
.
Parâmetros
índice de saída | o índice do tensor de saída desejado |
---|
Lança
IllegalStateException | se este modelo não contiver metadados de modelo |
---|
público MetadataExtractor.QuantizationParams getOutputTensorQuantizationParams (int outputIndex)
Obtém os parâmetros de quantização para o tensor de saída especificado por outputIndex
.
Parâmetros
índice de saída | o índice do tensor de saída desejado |
---|
public int[] getOutputTensorShape (int outputIndex)
Obtém a forma do tensor de saída com outputIndex
.
Parâmetros
índice de saída | o índice do tensor de saída desejado |
---|
byte público getOutputTensorType (int outputIndex)
Obtém o ERROR(/TensorType)
do tensor de saída com outputIndex
.
Parâmetros
índice de saída | o índice do tensor de saída desejado |
---|
hasMetadata booleano público ()
Retorna true
se o modelo tiver metadados. Caso contrário, retorna false
.
público final booleano isMinimumParserVersionSatisfied ()
Retorna true
se a versão mínima do analisador exigida pelo flatbuffer de metadados fornecido precede ou é igual à versão do analisador de metadados em que esta biblioteca MetadataExtractor depende. Todos os campos nos metadados podem ser analisados corretamente com esta biblioteca extratora de metadados neste caso. Caso contrário, ele retorna false
.
Por exemplo, suponha que a versão subjacente do analisador de metadados seja 1.14.1
,
- ele retorna
true
, se a versão mínima necessária do analisador for igual ou mais antiga, como1.14.1
ou1.14.0
. A versão nula precede todas as versões numéricas, porque alguns flatbuffers de metadados são gerados antes do primeiro lançamento versionado; - ele retorna
false
, se a versão mínima necessária do analisador for mais recente, como1.14.2
.