PymvDB

Pymvdb

Description

PyMVDB est une bibliothèque Python conçue pour créer et gérer une base de données vectorielle pour les images. Il est livré avec la possibilité d'utiliser des modèles d'extraction de fonctionnalités d'image de visage étreint comme cocoteuse.

L'exécution avec Google / Vit-Base-Patch16-224-in21k prend 0,41 seconde pour coder une image sur ma machine (Ryzen 7 2700)

Installation

Vous pouvez installer PYMVDB en utilisant PIP (à venir bientôt):

pip install pymvdb

En attendant que je le publie sur PIP, vous pouvez télécharger et installer PYMVDB en utilisant:

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

Utilisation locale

 # 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 )

Vous pouvez exécuter un exemple rapide en utilisant le fichier de départ. Toutes les images utilisées ici viennent de Wikipedia

python StartingPoint.py Test_car.jpg