MetadataExtractor

публичный класс MetadataExtractor

Загружает метаданные из TFLite Model FlatBuffer.

Модель TFLite FlatBuffer может быть создана с использованием файла схемы модели TFLite.

Некоторые модели содержат плоский буфер метаданных TFLite, в который записывает дополнительную информацию о том, что делает модель и как ее интерпретировать. Плоский буфер метаданных TFLite можно создать с помощью файла схемы метаданных TFLite.

Разрешено передавать модель FlatBuffer без метаданных TFLite. Однако вызов методов, считывающих метаданные TFLite, приведет к ошибкам во время выполнения.

Аналогичным образом разрешено передавать модель FlatBuffer без связанных файлов. Однако вызов методов, читающих связанные файлы, приведет к ошибкам во время выполнения.

Хотя модель TFLite FlatBuffer поддерживает несколько подграфов, TFLite Interpreter пока поддерживает только один подграф. Дополнительную информацию см. в инструкции по указанию подграфа во время преобразования. Поэтому MetadataExtractor опускает индекс подграфа в качестве входных данных в своих методах.

Вложенные классы

сорт MetadataExtractor.QuantizationParams Параметры квантования, соответствующие таблице QuantizationParameters в файле схемы модели TFLite.

Публичные конструкторы

MetadataExtractor (буфер ByteBuffer )
Создает MetadataExtractor с моделью TFLite FlatBuffer.

Публичные методы

Входной поток
getAssociatedFile ( строковое имя_файла)
Получает упакованный связанный файл с указанным fileName .
Установить < строка >
getAssociatedFileNames ()
Получает имена связанных файлов.
интервал
getInputTensorCount ()
Получает количество входных тензоров в модели.
Тензорные метаданные
getInputTensorMetadata (int inputIndex)
Получает метаданные для входного тензора, указанного inputIndex .
MetadataExtractor.QuantizationParams
getInputTensorQuantizationParams (int inputIndex)
Получает параметры квантования для входного тензора, заданного inputIndex .
интервал []
getInputTensorShape (int inputIndex)
Получает форму входного тензора с помощью inputIndex .
байт
getInputTensorType (int inputIndex)
Получает ERROR(/TensorType) входного тензора с помощью inputIndex .
Метаданные модели
getModelMetadata ()
Получает корневой обработчик метаданных модели.
интервал
getOutputTensorCount ()
Получает количество выходных тензоров в модели.
Тензорные метаданные
getOutputTensorMetadata (int outputIndex)
Получает метаданные для выходного тензора, указанного outputIndex .
MetadataExtractor.QuantizationParams
getOutputTensorQuantizationParams (int outputIndex)
Получает параметры квантования для выходного тензора, заданного параметром outputIndex .
интервал []
getOutputTensorShape (int выходной индекс)
Получает форму выходного тензора с помощью outputIndex .
байт
getOutputTensorType (int выходной индекс)
Получает ERROR(/TensorType) выходного тензора с помощью outputIndex .
логическое значение
имеетМетаданные ()
Возвращает true если модель имеет метаданные.
последнее логическое значение
isMinimumParserVersionSatisfied ()
Возвращает значение true если минимальная версия анализатора, требуемая для данного плоского буфера метаданных, предшествует или равна версии анализатора метаданных, от которой зависит эта библиотека MetadataExtractor.

Унаследованные методы

Публичные конструкторы

общедоступный MetadataExtractor (буфер ByteBuffer )

Создает MetadataExtractor с моделью TFLite FlatBuffer.

Параметры
буфер модель TFLite FlatBuffer
Броски
IllegalArgumentException если количество входных или выходных тензоров в модели не соответствует количеству в метаданных
Исключение IO если возникает ошибка при чтении модели в виде ZIP-файла

Публичные методы

public InputStream getAssociatedFile ( String fileName)

Получает упакованный связанный файл с указанным fileName .

Параметры
имя файла имя связанного файла
Возврат
  • необработанный входной поток, содержащий указанный файл
Броски
IllegalStateException если модель не является zip-файлом
IllegalArgumentException если указанный файл не существует в модели

public Set <String> getAssociatedFileNames ()

Получает имена связанных файлов.

Возврат
  • имена файлов связанных файлов
Броски
IllegalStateException если модель не является zip-файлом

public int getInputTensorCount ()

Получает количество входных тензоров в модели.

общедоступные TensorMetadata getInputTensorMetadata (int inputIndex)

Получает метаданные для входного тензора, указанного inputIndex .

Параметры
инпутиндекс индекс желаемого входного тензора
Броски
IllegalStateException если эта модель не содержит метаданных модели

public MetadataExtractor.QuantizationParams getInputTensorQuantizationParams (int inputIndex)

Получает параметры квантования для входного тензора, заданного inputIndex .

Параметры
инпутиндекс индекс желаемого входного тензора

public int[] getInputTensorShape (int inputIndex)

Получает форму входного тензора с помощью inputIndex .

Параметры
инпутиндекс индекс желаемого входного тензора

общедоступный байт getInputTensorType (int inputIndex)

Получает ERROR(/TensorType) входного тензора с помощью inputIndex .

Параметры
инпутиндекс индекс желаемого входного тензора

общедоступные ModelMetadata getModelMetadata ()

Получает корневой обработчик метаданных модели.

Броски
IllegalStateException если эта модель не содержит метаданных модели

public int getOutputTensorCount ()

Получает количество выходных тензоров в модели.

общедоступные TensorMetadata getOutputTensorMetadata (int outputIndex)

Получает метаданные для выходного тензора, указанного outputIndex .

Параметры
выходной индекс индекс желаемого выходного тензора
Броски
IllegalStateException если эта модель не содержит метаданных модели

public MetadataExtractor.QuantizationParams getOutputTensorQuantizationParams (int outputIndex)

Получает параметры квантования для выходного тензора, заданного параметром outputIndex .

Параметры
выходной индекс индекс желаемого выходного тензора

public int[] getOutputTensorShape (int outputIndex)

Получает форму выходного тензора с помощью outputIndex .

Параметры
выходной индекс индекс желаемого выходного тензора

общедоступный байт getOutputTensorType (int outputIndex)

Получает ERROR(/TensorType) выходного тензора с помощью outputIndex .

Параметры
выходной индекс индекс желаемого выходного тензора

общедоступное логическое значение hasMetadata ()

Возвращает true если модель имеет метаданные. В противном случае возвращает false .

публичное окончательное логическое значение isMinimumParserVersionSatisfied ()

Возвращает значение true если минимальная версия анализатора, требуемая для данного плоского буфера метаданных, предшествует или равна версии анализатора метаданных, от которой зависит эта библиотека MetadataExtractor. В этом случае все поля метаданных можно правильно проанализировать с помощью этой библиотеки извлечения метаданных. В противном случае он возвращает false .

Например, предположим, что базовая версия анализатора метаданных — 1.14.1 .

  • он возвращает true , если требуемая минимальная версия парсера такая же или старше, например 1.14.1 или 1.14.0 . Нулевая версия предшествует всем числовым версиям, поскольку некоторые плоские буферы метаданных создаются до первого выпуска версии;
  • он возвращает false , если требуемая минимальная версия парсера новее, например 1.14.2 .