Выполняет обнаружение объектов на изображениях.
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. |
Публичные методы
Унаследованные методы
Публичные методы
общедоступный статический 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
поддерживает следующие параметры:
- поворот изображения (через
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)
). По умолчанию используетсяImageProcessingOptions.Orientation.TOP_LEFT
.
Параметры
изображение | объект UINT8 TensorImage , который представляет изображение RGB или YUV |
---|---|
параметры | параметры настройки предварительной обработки изображения |
Броски
IllegalStateException | если произошла внутренняя ошибка |
---|---|
Исключение времени выполнения | если есть неуказанная ошибка |
IllegalArgumentException | если тип цветового пространства изображения не поддерживается |
общедоступный список < обнаружение > обнаружить (изображение MlImage , параметры ImageProcessingOptions )
Выполняет фактическое обнаружение предоставленного MlImage
с помощью ImageProcessingOptions
.
ObjectDetector
поддерживает следующие параметры:
- поворот изображения (через
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)
). По умолчанию используетсяImageProcessingOptions.Orientation.TOP_LEFT
.MlImage.getRotation()
неэффективен.
Параметры
изображение | объект MlImage , представляющий изображение |
---|---|
параметры | параметры настройки предварительной обработки изображения |
Броски
IllegalStateException | если произошла внутренняя ошибка |
---|---|
Исключение времени выполнения | если есть неуказанная ошибка |
IllegalArgumentException | если тип хранения или формат изображения не поддерживается |
общедоступный список < обнаружение > обнаружить (изображение TensorImage )
Выполняет фактическое обнаружение на предоставленном изображении.
ObjectDetector
поддерживает следующие типы цветового пространства TensorImage
:
Параметры
изображение | объект UINT8 TensorImage , который представляет изображение RGB или YUV |
---|
Броски
IllegalStateException | если произошла внутренняя ошибка |
---|---|
Исключение времени выполнения | если есть неуказанная ошибка |
IllegalArgumentException | если тип цветового пространства изображения не поддерживается |