py search images

Alat CLI: Pencarian gambar menggunakan klip dan vektor embeddings

Alat CLI ini memungkinkan Anda untuk mengindeks gambar dari folder yang ditentukan dan mencari gambar serupa berdasarkan kueri teks menggunakan klip (pretraining bahasa-citra kontras), pytorch dan numpy. Ini menghasilkan vektor gambar dan memungkinkan Anda untuk melakukan pencarian semantik secara efisien.

Fitur

• Indeks gambar dan menghasilkan embeddings vektor.
• Cari gambar berdasarkan kueri teks.
• Mendukung beberapa format gambar (.jpg, .jpeg, .png, .webp).

Penggunaan

Untuk mencari gambar dengan kueri teks, Anda harus terlebih dahulu mengindeks folder yang berisi gambar menggunakan alat ini.

Gambar indeks

Untuk mengindeks gambar dari images folder, jalankan:

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

Perintah ini akan memproses gambar di folder gambar, menghasilkan embeddings vektor mereka menggunakan model klip, dan menyimpannya ke vektor.npz di folder yang sama.

Cari gambar yang relevan

Untuk mencari gambar serupa berdasarkan kueri teks, jalankan:

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

Ganti "kueri teks Anda" dengan deskripsi gambar yang Anda cari. Alat ini akan memuat vektor dari vektor.npz dan mengembalikan gambar yang paling mirip.