py search images

Ferramenta CLI: Pesquisa de imagem usando incorporações de clipe e vetor

Esta ferramenta CLI permite indexar imagens de uma pasta especificada e pesquisar imagens semelhantes com base em uma consulta de texto usando clipe (pré-treinamento contrastivo de imagem de linguagem), Pytorch e Numpy. Ele gera vetores de imagem e permite que você realize pesquisas semânticas com eficiência.

Características

• Imagens de índice e gerar incorporações vetoriais.
• Pesquise imagens com base em consultas de texto.
• Suporta vários formatos de imagem (.jpg, .jpeg, .png, .webp).

Uso

Para pesquisar imagens por consulta de texto, você deve primeiro indexar a pasta que contém as imagens usando esta ferramenta.

Imagens de índice

Para indexar imagens de uma pasta images , execute:

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

Este comando processará as imagens na pasta de imagens, gerará suas incorporações de vetor usando o modelo de clipe e as salvará em vetores.npz na mesma pasta.

Procure imagens relevantes

Para procurar imagens semelhantes com base em uma consulta de texto, execute:

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

Substitua "sua consulta de texto" pela descrição da imagem que você está procurando. A ferramenta carregará os vetores de vetores.npz e retornará as imagens mais semelhantes.