py search images

Herramienta CLI: búsqueda de imágenes usando incrustaciones de clip e vector

Esta herramienta CLI le permite indexar imágenes desde una carpeta especificada y buscar imágenes similares basadas en una consulta de texto usando clip (pretrenesa de imagen de lenguaje contrastante), pytorch y numpy. Genera vectores de imagen y le permite realizar búsquedas semánticas de manera eficiente.

Características

• Index imágenes y generar integridades vectoriales.
• Busque imágenes basadas en consultas de texto.
• Admite múltiples formatos de imagen (.jpg, .jpeg, .png, .webp).

Uso

Para buscar imágenes por consulta de texto, primero debe indexar la carpeta que contiene las imágenes utilizando esta herramienta.

Índice de imágenes

Para indexar imágenes desde una carpeta images , ejecute:

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

Este comando procesará las imágenes en la carpeta de imágenes, generará sus incrustaciones vectoriales usando el modelo de clip y las guardará en vectores.npz en la misma carpeta.

Buscar imágenes relevantes

Para buscar imágenes similares basadas en una consulta de texto, ejecute:

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

Reemplace "su consulta de texto" con la descripción de la imagen que está buscando. La herramienta cargará los vectores de vectores.npz y devolverá las imágenes más similares.