Image to Image Search

SmartSearch - это поисковая система обратного изображения, которая находит аналогичные изображения, генерируя подписи и сравнивая эти подписи. Основано Tensorflow и Elasticsearch.

Обновление : зеркало этого репозитория в Codeberg: Image-Image-Search

ОБНОВЛЕНИЕ : REST API для генератора подписи изображения и поиска изображений добавлены! (Крик к XC0D3RZ)

Демо

Глубокая поисковая система обратного изображения - YouTube демонстрируют основную функциональность.

Ветви

Docker Image

Основная реализация

Пакеты требуются:

• Анаконда
• Кера с бэкэнд с тензорфлоу (Python 3.6)
• Elastic Search и Elasticsearch-Py (Elastic Search 6.0)

Для получения дополнительной информации проверьте требования.

Предварительно обученные модели

• Flickr-8k LSTM Weights (flickr8k_cnn_lstm_v1.p)

Загрузите это и вставьте его в папку моделей.

Выход

Советы

• Установите Elasticsearch и всегда проверяйте, работает ли процесс эластичного поиска перед запуском server.py или index_database.py.

• Вместо того, чтобы использовать функциональность загрузки, вставьте все ваши изображения в папку static/img а затем python index_database.py для индексации всех этих изображений.

• Если вы хотите удалить индексированные изображения, DO sh delete_index.sh

Спасибо

Благодаря материализованию CSS за эстетику, красивую напечатанную для учебников Flask и Andrej Karpathy для замечательной библиотеки подписи. Выкрикивайте XC0D3RZ для остального включения API!