py search images

Outil CLI: recherche d'image à l'aide de Clip et Vector Imbeddings

Cet outil CLI vous permet d'indexer des images à partir d'un dossier spécifié et de rechercher des images similaires basées sur une requête de texte à l'aide de CLIP (pré-formation de langage contrastif), Pytorch et Numpy. Il génère des vecteurs d'image et vous permet d'effectuer efficacement les recherches sémantiques.

Caractéristiques

• Index des images et générer des intérêts vectoriels.
• Recherchez des images basées sur des requêtes de texte.
• Prend en charge plusieurs formats d'image (.jpg, .jpeg, .png, .webp).

Usage

Pour rechercher des images par requête de texte, vous devez d'abord indexer le dossier contenant les images à l'aide de cet outil.

Index des images

Pour indexer les images à partir d'un dossier images , exécutez:

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

Cette commande traitera les images dans le dossier Images, génèrera leurs intégres vectoriels à l'aide du modèle de clip et les enregistre sur vecteurs.npz dans le même dossier.

Rechercher des images pertinentes

Pour rechercher des images similaires basées sur une requête de texte, exécutez:

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

Remplacez "votre requête de texte" par la description de l'image que vous recherchez. L'outil chargera les vecteurs à partir de vecteurs.npz et renverra les images les plus similaires.