PymvDB

Pymvdb

Beschreibung

PYMVDB ist eine Python -Bibliothek, mit der eine Vektor -Datenbank für Bilder erstellt und verwaltet wird. Es wird mit der Möglichkeit verbunden, die Extraktionsmodelle für Gesichtsbild -Feature -Extraktion als Encoder zu verwenden.

Das Ausführen mit Google/Vit-Base-Patch16-224-in21K benötigt 0,41 Sekunden, um ein Bild auf meiner Maschine zu codieren (Ryzen 7 2700).

Installation

Sie können PYMVDB mit PIP (in Kürze) installieren:

pip install pymvdb

Während ich darauf wartete, dass ich es auf PIP veröffentlichen kann, können Sie PYMVDB mit:

git clone https://github.com/BBurgarella/PymvDB.git
cd PymvDB
pip install . 

Lokale Nutzung

 # Initialize the client with an embedding model
embedding_model = YourEmbeddingModel ()  # Replace with your actual embedding model
db = Client ( embedding_model , persistent_path = 'database.sqlite' )

# Create a new collection
collection = db . create_collection ( Name = 'my_collection' )

# Add an image to the collection
collection . add_image ( 'path/to/image' , metadata = { "..." })

# Find similar images
target_image = Image . open ( 'path/to/target_image' )
similar_images = collection . find_similar_images ( target_image , top_N = 5 )
print ( similar_images )

Sie können ein kurzes Beispiel mit der Startpunktdatei ausführen. Alle hier verwendeten Bilder stammen aus Wikipedia

python StartingPoint.py Test_car.jpg