MetadataExtractor

classe pública MetadataExtractor

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

MetadataExtractor (buffer ByteBuffer )
Cria um MetadataExtractor com modelo TFLite FlatBuffer.

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, como 1.14.1 ou 1.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, como 1.14.2 .