ObjectDetector

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

Выполняет обнаружение объектов на изображениях.

API ожидает модель TFLite с метаданными модели TFLite. .

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

  • Тензор входного изображения ( kTfLiteUInt8 / kTfLiteFloat32 )
    • Ввод изображения размера [batch x height x width x channels] .
    • Пакетный вывод не поддерживается ( batch должен быть равен 1).
    • поддерживаются только входы RGB ( channels должно быть 3).
    • если тип — kTfLiteFloat32 , то NormalizationOptions необходимо прикрепить к метаданным для нормализации ввода.
  • Выходные тензоры должны быть четырьмя выходами операции DetectionPostProcess , т.е.:
    • Тензор местоположения ( kTfLiteFloat32 ):
      • тензор размера [1 x num_results x 4] , внутренний массив, представляющий ограничивающие рамки в форме [сверху, слева, справа, снизу].
      • BoundingBoxProperties должны быть прикреплены к метаданным и должны указывать type=BOUNDARIES и coordinate_type=RATIO .
    • Тензор классов ( kTfLiteFloat32 ):
      • тензор размера [1 x num_results] , каждое значение представляет собой целочисленный индекс класса.
      • если карты меток прикреплены к метаданным как связанные файлы TENSOR_VALUE_LABELS , они используются для преобразования значений тензора в метки.
    • Тензор оценок ( kTfLiteFloat32 ):
      • тензор размера [1 x num_results] , каждое значение представляет оценку обнаруженного объекта.
    • Тензор количества обнаружения ( kTfLiteFloat32 ):
      • целое число num_results как тензор размера [1] .

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

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

сорт ObjectDetector.ObjectDetectorOptions Варианты настройки ObjectDetector.

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

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

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

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

общедоступный статический ObjectDetector createFromBuffer ( ByteBuffer modelBuffer)

Создает экземпляр ObjectDetector с буфером модели и параметром ObjectDetector.ObjectDetectorOptions по умолчанию.

Параметры
модельBuffer прямой ByteBuffer или MappedByteBuffer модели обнаружения
Броски
IllegalArgumentException если буфер модели не является прямым ByteBuffer или MappedByteBuffer * @throws IllegalStateException, если возникла внутренняя ошибка
Исключение времени выполнения если есть неуказанная ошибка

общедоступный статический ObjectDetector createFromBufferAndOptions (опции ByteBuffer modelBuffer, ObjectDetector.ObjectDetectorOptions )

Создает экземпляр ObjectDetector с буфером модели и ObjectDetector.ObjectDetectorOptions .

Параметры
модельBuffer прямой ByteBuffer или MappedByteBuffer модели обнаружения
параметры
Броски
IllegalArgumentException если буфер модели не является прямым ByteBuffer или MappedByteBuffer
IllegalStateException если произошла внутренняя ошибка
Исключение времени выполнения если есть неуказанная ошибка

общедоступный статический ObjectDetector createFromFile (контекстный контекст, String modelPath)

Создает экземпляр ObjectDetector на основе ObjectDetector.ObjectDetectorOptions по умолчанию.

Параметры
контекст
модельпуть путь к модели обнаружения с метаданными в активах
Броски
Исключение IO если при загрузке модели tflite возникает ошибка ввода-вывода
IllegalArgumentException если аргумент недействителен
IllegalStateException если произошла внутренняя ошибка
Исключение времени выполнения если есть неуказанная ошибка

общедоступный статический ObjectDetector createFromFile ( File modelFile)

Создает экземпляр ObjectDetector на основе ObjectDetector.ObjectDetectorOptions по умолчанию.

Параметры
МодельФайл Модель обнаружения Экземпляр File
Броски
Исключение IO если при загрузке модели tflite возникает ошибка ввода-вывода
IllegalArgumentException если аргумент недействителен
IllegalStateException если произошла внутренняя ошибка
Исключение времени выполнения если есть неуказанная ошибка

общедоступный статический ObjectDetector createFromFileAndOptions (контекст контекста, String modelPath, параметры ObjectDetector.ObjectDetectorOptions )

Создает экземпляр ObjectDetector из ObjectDetector.ObjectDetectorOptions .

Параметры
контекст
модельпуть путь к модели обнаружения с метаданными в активах
параметры
Броски
Исключение IO если при загрузке модели tflite возникает ошибка ввода-вывода
IllegalArgumentException если аргумент недействителен
IllegalStateException если произошла внутренняя ошибка
Исключение времени выполнения если есть неуказанная ошибка

общедоступный статический ObjectDetector createFromFileAndOptions (опции File modelFile, ObjectDetector.ObjectDetectorOptions )

Создает экземпляр ObjectDetector из ObjectDetector.ObjectDetectorOptions .

Параметры
МодельФайл Модель обнаружения Экземпляр File
параметры
Броски
Исключение IO если при загрузке модели tflite возникает ошибка ввода-вывода
IllegalArgumentException если аргумент недействителен
IllegalStateException если произошла внутренняя ошибка
Исключение времени выполнения если есть неуказанная ошибка

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

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

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

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

Выполняет фактическое обнаружение на предоставленном изображении.

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

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

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

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

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

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

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

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

Выполняет фактическое обнаружение на предоставленном изображении.

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

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