py search images

CLI -Tool: Bildsuche mit Clip- und Vektor -Einbettungen

Mit diesem CLI-Tool können Sie Bilder aus einem angegebenen Ordner aus indexieren und nach ähnlichen Bildern basierend auf einer Textabfrage unter Verwendung von Clip (kontrastive Sprachbild-Vorbereitung), Pytorch und Numpy suchen. Es generiert Bildvektoren und ermöglicht es Ihnen, semantische Suche effizient durchzuführen.

Merkmale

• Indexbilder und generieren Vektoreinbettungen.
• Suche nach Bildern basierend auf Textanfragen.
• Unterstützt mehrere Bildformate (.jpg, .jpeg, .png, .webp).

Verwendung

Um nach Textabfragen nach Bildern zu suchen, müssen Sie zuerst den Ordner mit diesem Tool indexieren, der die Bilder enthält.

Indexbilder

Um Bilder aus einem images zu indexieren, führen Sie aus:

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

Dieser Befehl verarbeitet die Bilder im Bilderordner im Bilder, generiert ihre Vektor -Einbettungen mithilfe des Clip -Modells und speichern Sie sie in Vektoren.npz im selben Ordner.

Suche nach relevanten Bildern

Um auf einer Textabfrage nach ähnlichen Bildern zu suchen, rennen Sie:

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

Ersetzen Sie "Ihre Textabfrage" durch die Beschreibung des Bildes, das Sie suchen. Das Tool lädt die Vektoren von Vektoren.npz und gibt die ähnlichsten Bilder zurück.