Image to Image Search

SmartSearch adalah mesin pencari gambar terbalik yang menemukan gambar serupa dengan menghasilkan teks dan membandingkan keterangan tersebut. Didukung oleh TensorFlow dan Elasticsearch.

UPDATE : Mirror Repositori ini di CodeBerg: Image-to-Image-Search

UPDATE : REST API untuk generator caption gambar dan pencarian gambar ditambahkan! (Shoutout ke XC0D3RZ)

Demo

Mesin pencari gambar terbalik yang dalam - Demo YouTube Fungsionalitas inti.

Cabang

Gambar Docker

Implementasi Dasar

Paket Diperlukan:

• Anaconda
• Keras dengan backend tensorflow (Python 3.6)
• Pencarian elastis dan Elasticsearch-Py (elastis pencarian 6.0)

Untuk lebih lanjut, lihat persyaratan.txt

Model pra-terlatih

• Flickr-8K LSTM Bobot (flickr8k_cnn_lstm_v1.p)

Unduh ini dan tempel di dalam folder model.

Keluaran

Tips

• Instal Elasticsearch dan selalu periksa apakah proses pencarian elastis berjalan sebelum meluncurkan server.py atau index_database.py.

• Alih -alih menggunakan fungsionalitas unggah, tempel semua gambar Anda di dalam folder static/img diikuti oleh python index_database.py untuk mengindeks semua gambar tersebut.

• Jika Anda ingin menghapus gambar yang diindeks, lakukan sh delete_index.sh

Terima kasih

Berkat CSS yang terwujud untuk estetika, dicetak cukup untuk tutorial Flask dan Andrej Karpathy untuk perpustakaan teks yang indah. Shoutout ke XC0D3RZ untuk inklusi API REST!