sis

Простая поисковая система изображения

Демо

Рабочий процесс

Новости

• [2020.06] Обновил многие части кода для учебника CVPR 2020

Обзор

• Простая поисковая система изображения на основе изображения с использованием керас + колба. Вы можете запустить поисковую систему, просто запустив два сценария Python.
• offline.py : Этот скрипт извлекает глубокую функцию из каждого изображения базы данных. Каждая функция представляет собой активацию FC6 4096D из модели VGG16 с предварительно обученными весами ImageNet.
• server.py : этот скрипт запускает веб-сервер. Вы можете отправить изображение запроса на сервер через Flask Web-Interface. Сервер находит аналогичные изображения с запросом с помощью простого линейного сканирования.
• Графические процессоры не требуются.
• Протестировано на Ubuntu 18.04 и WSL2 (Ubuntu 20.04)

Ссылки

• Демо
• Курс на CVPR2020 [слайды] [Видео]
• Project Page
• Учебник и видео от @sdkcarlos

Использование

git clone https://github.com/matsui528/sis.git
cd sis
pip install -r requirements.txt

# Put your image files (*.jpg) on static/img

# Then fc6 features are extracted and saved on static/feature
# Note that it takes time for the first time because Keras downloads the VGG weights.
python offline.py

# Now you can do the search via localhost:5000
python server.py

Advanced: запуск на AWS

• Вы можете легко запустить сервер поисковых систем на AWS EC2. Пожалуйста, сначала откройте порт 5000 и запустите экземпляр EC2. Обратите внимание, что вам нужно создать группу безопасности так, чтобы открылся порт 5000.
• Достаточный экземпляр ЦП среднего уровня, например, m5.large.
• После того, как вы войдите в экземпляр SSH, пожалуйста, настройте среду Python (например, Anaconda).
• Запустить offline.py и server.py .
• После запуска python server.py вы можете получить доступ к серверу из своего браузера через что-то вроде http://ec2-XX-XX-XXX-XXX.us-west-2.compute.amazonaws.com:5000
• (Advanced) Если вы хотите безопасно развернуть систему, пожалуйста, рассмотрите возможность запустить поисковую систему с обычным веб -сервером, например, UWSGI + NGINX.
• (Advanced) Если вы хотите развернуть системную без сервера, AWS Apprunner - это путь.

Цитирование

 @misc{sis,
    author = {Yusuke Matsui},
    title = {Simple Image Search Engine},
    howpublished = {url{https://github.com/matsui528/sis}}
}