PymvDB

Pymvdb

Keterangan

PYMVDB adalah perpustakaan Python yang dirancang untuk membuat dan mengelola database vektor untuk gambar. Muncul dengan kemampuan untuk menggunakan model ekstraksi fitur gambar pemeluk wajah sebagai encoder.

Menjalankannya dengan Google/VIT-BASE-PATCH16-224-IN21K membutuhkan 0,41 detik untuk menyandikan gambar pada mesin saya (Ryzen 7 2700)

Instalasi

Anda dapat menginstal PYMVDB menggunakan PIP (segera hadir):

pip install pymvdb

Sambil menunggu saya untuk mempublikasikannya di PIP, Anda dapat mengunduh dan menginstal PYMVDB menggunakan:

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

Penggunaan lokal

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

Anda dapat menjalankan contoh cepat dengan menggunakan file StartingPoint. Semua gambar yang digunakan di sini berasal dari Wikipedia

python StartingPoint.py Test_car.jpg