ImageClassifier

публичный финальный класс ImageClassifier

Выполняет классификацию изображений.

API ожидает модель TFLite с необязательными, но настоятельно рекомендуемыми метаданными модели TFLite. .

API поддерживает модели с одним входным тензором изображения и одним выходным тензором классификации. Если быть более конкретным, то вот требования.

  • Тензор входного изображения ( kTfLiteUInt8 / kTfLiteFloat32 )
    • Ввод изображения размера [batch x height x width x channels] .
    • Пакетный вывод не поддерживается ( batch должен быть равен 1).
    • поддерживаются только входы RGB ( channels должно быть 3).
    • если тип — kTfLiteFloat32 , то NormalizationOptions необходимо прикрепить к метаданным для нормализации ввода.
  • Тензор выходной оценки ( kTfLiteUInt8 / kTfLiteFloat32 )

Пример такой модели можно найти на TensorFlow Hub. .

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

сорт ImageClassifier.ImageClassifierOptions Варианты настройки ImageClassifier.

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

Список < Классификации >
классифицировать (изображение TensorImage )
Выполняет фактическую классификацию предоставленного TensorImage .
Список < Классификации >
классифицировать (изображение TensorImage , параметры ImageProcessingOptions )
Выполняет фактическую классификацию предоставленного TensorImage с помощью ImageProcessingOptions .
Список < Классификации >
классифицировать (изображение MlImage )
Выполняет фактическую классификацию предоставленного MlImage .
Список < Классификации >
классифицировать (изображение MlImage , параметры ImageProcessingOptions )
Выполняет фактическую классификацию предоставленного MlImage с помощью ImageProcessingOptions .
статический классификатор изображений
createFromBuffer ( ByteBuffer modelBuffer)
Создает экземпляр ImageClassifier с буфером модели и параметром ImageClassifier.ImageClassifierOptions по умолчанию.
статический классификатор изображений
createFromBufferAndOptions (опции ByteBuffer modelBuffer, ImageClassifier.ImageClassifierOptions )
Создает экземпляр ImageClassifier с буфером модели и ImageClassifier.ImageClassifierOptions .
статический классификатор изображений
createFromFile (контекстный контекст, String modelPath)
Создает экземпляр ImageClassifier на основе ImageClassifier.ImageClassifierOptions по умолчанию.
статический классификатор изображений
createFromFile ( Файл модели файла)
Создает экземпляр ImageClassifier на основе ImageClassifier.ImageClassifierOptions по умолчанию.
статический классификатор изображений
createFromFileAndOptions (контекст контекста, String modelPath, параметры ImageClassifier.ImageClassifierOptions )
Создает экземпляр ImageClassifier из ImageClassifier.ImageClassifierOptions .
статический классификатор изображений
createFromFileAndOptions (параметры File modelFile, ImageClassifier.ImageClassifierOptions )
Создает экземпляр ImageClassifier .

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

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

общедоступный список < Классификации > классифицировать (изображение TensorImage )

Выполняет фактическую классификацию предоставленного TensorImage .

ImageClassifier поддерживает следующие типы цветового пространства TensorImage :

Параметры
изображение объект UINT8 TensorImage , который представляет изображение RGB или YUV
Броски
IllegalArgumentException если тип цветового пространства изображения не поддерживается

общедоступный список < Классификации > классифицировать (изображение TensorImage , параметры ImageProcessingOptions )

Выполняет фактическую классификацию предоставленного TensorImage с помощью ImageProcessingOptions .

ImageClassifier поддерживает следующие параметры:

ImageClassifier поддерживает следующие типы цветового пространства TensorImage :

Параметры
изображение объект UINT8 TensorImage , который представляет изображение RGB или YUV
параметры
Броски
IllegalArgumentException если тип цветового пространства изображения не поддерживается

общедоступный список < Классификации > классифицировать (изображение MlImage )

Выполняет фактическую классификацию предоставленного MlImage .

Параметры
изображение объект MlImage , представляющий изображение
Броски
IllegalArgumentException если тип хранения или формат изображения не поддерживается

общедоступный список < Классификации > классифицировать (изображение MlImage , параметры ImageProcessingOptions )

Выполняет фактическую классификацию предоставленного MlImage с помощью ImageProcessingOptions .

ImageClassifier поддерживает следующие параметры:

Параметры
изображение объект 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 если произошла внутренняя ошибка
Исключение времени выполнения если есть неуказанная ошибка