Выполняет классификацию изображений.
API ожидает модель TFLite с необязательными, но настоятельно рекомендуемыми метаданными модели TFLite. .
API поддерживает модели с одним входным тензором изображения и одним выходным тензором классификации. Если быть более конкретным, то вот требования.
- Тензор входного изображения (
kTfLiteUInt8
/kTfLiteFloat32
)- Ввод изображения размера
[batch x height x width x channels]
. - Пакетный вывод не поддерживается (
batch
должен быть равен 1). - поддерживаются только входы RGB (
channels
должно быть 3). - если тип —
kTfLiteFloat32
, то NormalizationOptions необходимо прикрепить к метаданным для нормализации ввода.
- Ввод изображения размера
kTfLiteUInt8
/ kTfLiteFloat32
)- с
N
классами 2 или 4 измерений, например[1 x N]
или[1 x 1 x 1 x N]
- файл этикетки необходимо упаковать в метаданные. См. пример создания метаданных для классификатора изображений . Если файлы меток не упакованы, в качестве метки в результате будет использоваться индекс.
Пример такой модели можно найти на TensorFlow Hub. .
Вложенные классы
сорт | ImageClassifier.ImageClassifierOptions | Варианты настройки ImageClassifier. |
Публичные методы
Унаследованные методы
Публичные методы
общедоступный список < Классификации > классифицировать (изображение TensorImage )
Выполняет фактическую классификацию предоставленного TensorImage
.
ImageClassifier
поддерживает следующие типы цветового пространства TensorImage
:
Параметры
изображение | объект UINT8 TensorImage , который представляет изображение RGB или YUV |
---|
Броски
IllegalArgumentException | если тип цветового пространства изображения не поддерживается |
---|
общедоступный список < Классификации > классифицировать (изображение TensorImage , параметры ImageProcessingOptions )
Выполняет фактическую классификацию предоставленного TensorImage
с помощью ImageProcessingOptions
.
ImageClassifier
поддерживает следующие параметры:
- Область интереса (ROI) (через
ImageProcessingOptions.Builder.setRoi(Rect)
). По умолчанию используется все изображение. - поворот изображения (через
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)
). По умолчанию используетсяImageProcessingOptions.Orientation.TOP_LEFT
.
ImageClassifier
поддерживает следующие типы цветового пространства TensorImage
:
Параметры
изображение | объект UINT8 TensorImage , который представляет изображение RGB или YUV |
---|---|
параметры |
Броски
IllegalArgumentException | если тип цветового пространства изображения не поддерживается |
---|
общедоступный список < Классификации > классифицировать (изображение MlImage )
Выполняет фактическую классификацию предоставленного MlImage
.
Параметры
изображение | объект MlImage , представляющий изображение |
---|
Броски
IllegalArgumentException | если тип хранения или формат изображения не поддерживается |
---|
общедоступный список < Классификации > классифицировать (изображение MlImage , параметры ImageProcessingOptions )
Выполняет фактическую классификацию предоставленного MlImage
с помощью ImageProcessingOptions
.
ImageClassifier
поддерживает следующие параметры:
- Область интереса (ROI) (через
ImageProcessingOptions.Builder.setRoi(Rect)
). По умолчанию используется все изображение. - поворот изображения (через
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)
). По умолчанию используетсяImageProcessingOptions.Orientation.TOP_LEFT
.MlImage.getRotation()
неэффективен.
Параметры
изображение | объект MlImage , представляющий изображение |
---|---|
параметры | настраивает параметры, включая рентабельность инвестиций и вращение |
Броски
IllegalArgumentException | если тип хранения или формат изображения не поддерживается |
---|
общедоступный статический ImageClassifier createFromBuffer ( ByteBuffer modelBuffer)
Создает экземпляр ImageClassifier
с буфером модели и параметром ImageClassifier.ImageClassifierOptions
по умолчанию.
Параметры
модельBuffer | прямой ByteBuffer или MappedByteBuffer модели классификации |
---|
Броски
IllegalArgumentException | если буфер модели не является прямым ByteBuffer или MappedByteBuffer |
---|---|
IllegalStateException | если произошла внутренняя ошибка |
Исключение времени выполнения | если есть неуказанная ошибка |
общедоступный статический ImageClassifier createFromBufferAndOptions (опции ByteBuffer modelBuffer, ImageClassifier.ImageClassifierOptions )
Создает экземпляр ImageClassifier
с буфером модели и ImageClassifier.ImageClassifierOptions
.
Параметры
модельBuffer | прямой ByteBuffer или MappedByteBuffer модели классификации |
---|---|
параметры |
Броски
IllegalArgumentException | если буфер модели не является прямым ByteBuffer или MappedByteBuffer |
---|---|
IllegalStateException | если произошла внутренняя ошибка |
Исключение времени выполнения | если есть неуказанная ошибка |
общедоступный статический ImageClassifier createFromFile (контекст контекста, String modelPath)
Создает экземпляр ImageClassifier
на основе ImageClassifier.ImageClassifierOptions
по умолчанию.
Параметры
контекст | |
---|---|
модельпуть | путь модели классификации с метаданными в активах |
Броски
Исключение IO | если при загрузке модели tflite возникает ошибка ввода-вывода |
---|---|
IllegalArgumentException | если аргумент недействителен |
IllegalStateException | если произошла внутренняя ошибка |
Исключение времени выполнения | если есть неуказанная ошибка |
public static ImageClassifier createFromFile ( File modelFile)
Создает экземпляр ImageClassifier
на основе ImageClassifier.ImageClassifierOptions
по умолчанию.
Параметры
МодельФайл | экземпляр File модели классификации |
---|
Броски
Исключение IO | если при загрузке модели tflite возникает ошибка ввода-вывода |
---|---|
IllegalArgumentException | если аргумент недействителен |
IllegalStateException | если произошла внутренняя ошибка |
Исключение времени выполнения | если есть неуказанная ошибка |
общедоступный статический ImageClassifier createFromFileAndOptions (контекст контекста, String modelPath, параметры ImageClassifier.ImageClassifierOptions )
Создает экземпляр ImageClassifier
из ImageClassifier.ImageClassifierOptions
.
Параметры
контекст | |
---|---|
модельпуть | путь модели классификации с метаданными в активах |
параметры |
Броски
Исключение IO | если при загрузке модели tflite возникает ошибка ввода-вывода |
---|---|
IllegalArgumentException | если аргумент недействителен |
IllegalStateException | если произошла внутренняя ошибка |
Исключение времени выполнения | если есть неуказанная ошибка |
общедоступный статический ImageClassifier createFromFileAndOptions (опции File modelFile, ImageClassifier.ImageClassifierOptions )
Создает экземпляр ImageClassifier
.
Параметры
МодельФайл | экземпляр File модели классификации |
---|---|
параметры |
Броски
Исключение IO | если при загрузке модели tflite возникает ошибка ввода-вывода |
---|---|
IllegalArgumentException | если аргумент недействителен |
IllegalStateException | если произошла внутренняя ошибка |
Исключение времени выполнения | если есть неуказанная ошибка |