tflite_support.task.vision.ImageSearcher

Class to performs image search.

tflite_support.task.vision.ImageSearcher(
    options: tflite_support.task.vision.ImageSearcherOptions,
    cpp_searcher: _CppImageSearcher
) -> None

It works by performing embedding extraction on images, followed by nearest-neighbor search in an index of embeddings through ScaNN.

options

Methods

create_from_file

View source

@classmethod
create_from_file(
    model_file_path: str, index_file_path: Optional[str] = None
) -> 'ImageSearcher'

Creates the ImageSearcher object from a TensorFlow Lite model.

Args
model_file_path Path to the model.
index_file_path Path to the index. Only required if the index is not attached to the output tensor metadata as an AssociatedFile with type SCANN_INDEX_FILE.

Returns
ImageSearcher object that's created from options.

Raises
ValueError If failed to create ImageSearcher object from the provided file such as invalid file.
RuntimeError If other types of error occurred.

create_from_options

View source

@classmethod
create_from_options(
    options: tflite_support.task.vision.ImageSearcherOptions
) -> 'ImageSearcher'

Creates the ImageSearcher object from image searcher options.

Args
options Options for the image searcher task.

Returns
ImageSearcher object that's created from options.

Raises
ValueError If failed to create ImageSearcher object from ImageSearcherOptions such as missing the model.
RuntimeError If other types of error occurred.

get_user_info

View source