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加載向量並返回最相似的圖像。