Посетите симпозиум «Женщины в машинном обучении» 7 декабря Зарегистрируйтесь сейчас

ObjectDetector

общественный конечный класс ObjectDetector

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

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

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

  • Входной тензор изображения ( kTfLiteUInt8 / kTfLiteFloat32 )
    • входное изображение размера [batch x height x width x channels] .
    • пакетный вывод не поддерживается ( batch требуется , чтобы быть 1).
    • Только RGB входы поддерживаются ( channels требуется , чтобы быть 3).
    • если тип является kTfLiteFloat32 , NormalizationOptions должны быть прикреплены к метаданным для ввода нормализации.
  • Выходные тензоры должны быть 4 выходов в 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.

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

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

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

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

общественности статической ObjectDetector createFromBuffer ( ByteBuffer modelBuffer)

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

Параметры
modelBuffer прямой ByteBuffer или MappedByteBuffer модели обнаружения
Броски
IllegalArgumentException если модель буфер не является прямым ByteBuffer или А MappedByteBuffer * @throws IllegalStateException если есть внутренняя ошибка
RuntimeException если есть иная неопределенная ошибка

общественности статической ObjectDetector createFromBufferAndOptions ( ByteBuffer modelBuffer, ObjectDetector.ObjectDetectorOptions варианты)

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

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

общественности статической ObjectDetector createFromFile (контекст Контекст, строка modelPath)

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

Параметры
контекст
modelPath путь к модели обнаружения с метаданными в активах
Броски
IOException если при загрузке модели tflite возникает ошибка ввода / вывода
IllegalArgumentException если аргумент недействителен
IllegalStateException если есть внутренняя ошибка
RuntimeException если есть иная неопределенная ошибка

общественности статической ObjectDetector createFromFile ( Файл ModelFile)

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

Параметры
modelFile обнаружение модель File экземпляр
Броски
IOException если при загрузке модели tflite возникает ошибка ввода / вывода
IllegalArgumentException если аргумент недействителен
IllegalStateException если есть внутренняя ошибка
RuntimeException если есть иная неопределенная ошибка

общественности статической ObjectDetector createFromFileAndOptions (контекст Контекст, строка modelPath, ObjectDetector.ObjectDetectorOptions опции)

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

Параметры
контекст
modelPath путь к модели обнаружения с метаданными в активах
параметры
Броски
IOException если при загрузке модели tflite возникает ошибка ввода / вывода
IllegalArgumentException если аргумент недействителен
IllegalStateException если есть внутренняя ошибка
RuntimeException если есть иная неопределенная ошибка

общественности статической ObjectDetector createFromFileAndOptions ( Файл ModelFile, ObjectDetector.ObjectDetectorOptions опция)

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

Параметры
modelFile обнаружение модель File экземпляр
параметры
Броски
IOException если при загрузке модели tflite возникает ошибка ввода / вывода
IllegalArgumentException если аргумент недействителен
IllegalStateException если есть внутренняя ошибка
RuntimeException если есть иная неопределенная ошибка

общественности Список < Detection > обнаружения ( MlImage изображение)

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

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

общественности Список < Detection > обнаружения ( TensorImage изображения, ImageProcessingOptions варианты)

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

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

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

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

общественности Список < Detection > обнаружения ( MlImage изображения, ImageProcessingOptions варианты)

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

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

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

общественности Список < Detection > обнаружения ( TensorImage изображение)

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

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

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