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