py search images

CLI工具：使用剪辑和向量嵌入的图像搜索

该CLI工具使您可以使用指定文件夹索引图像，并使用剪辑（对比度语言图像预处理），Pytorch和Numpy搜索基于文本查询的相似图像。它生成图像向量，并允许您有效地执行语义搜索。

特征

• 索引图像并生成向量嵌入。
• 根据文本查询搜索图像。
• 支持多个图像格式（.jpg，.jpeg，.png，.webp）。

用法

要通过文本查询搜索图像，您必须首先使用此工具索引包含图像的文件夹。

索引图像

要从文件夹图像索引images ，请运行：

(venv) ➜ vector-embeddings python main.py index -f images

此命令将处理图像文件夹中的图像，使用剪辑模型生成其向量嵌入，然后将它们保存到同一文件夹中的vectors.npz。

搜索相关图像

要根据文本查询搜索类似的图像，请运行：

(venv) ➜ vector-embeddings python main.py search " Your text query " -f images

将“文本查询”替换为您要查找的图像的描述。该工具将从vectors.npz加载向量并返回最相似的图像。