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
ImageSearche