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.
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