sis

Moteur de recherche d'images simples

Démo

Flux de travail

Nouvelles

• [2020.06] Mise à jour de nombreuses parties du code pour le tutoriel CVPR 2020

Aperçu

• Moteur de recherche d'image simple basé sur l'image utilisant Keras + Flask. Vous pouvez lancer le moteur de recherche simplement en exécutant deux scripts Python.
• offline.py : Ce script extrait une fonction profonde de chaque image de base de données. Chaque caractéristique est une activation 4096D FC6 à partir d'un modèle VGG16 avec des poids pré-entraînés ImageNet.
• server.py : ce script exécute un serveur Web. Vous pouvez envoyer votre image de requête au serveur via une interface Web FLASK. Le serveur trouve des images similaires à la requête par une simple analyse linéaire.
• Les GPU ne sont pas nécessaires.
• Testé sur Ubuntu 18.04 et WSL2 (Ubuntu 20.04)

Links

• Démo
• Cours à CVPR2020 [diapositives] [vidéo]
• Page du projet
• Tutoriel et vidéo de @sdkcarlos

Usage

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: Location sur AWS

• Vous pouvez facilement lancer le serveur de moteur de recherche sur AWS EC2. Veuillez d'abord ouvrir le port 5000 et lancer une instance EC2. Notez que vous devez créer un groupe de sécurité de telle sorte que le port 5000 soit ouvert.
• Une instance de processeur de niveau intermédiaire est suffisante, par exemple, m5.lage.
• Après vous être connecté à l'instance par SSH, veuillez configurer l'environnement Python (par exemple, par Anaconda).
• Exécutez offline.py et server.py .
• Après avoir exécuté python server.py , vous pouvez accéder au serveur à partir de votre navigateur via quelque chose comme http://ec2-XX-XX-XXX-XXX.us-west-2.compute.amazonaws.com:5000
• (Advanced) Si vous souhaitez déployer le système de manière sécurisée, veuillez envisager d'exécuter le moteur de recherche avec le serveur Web habituel, par exemple, UWSGI + NGINX.
• (Advanced) Si vous souhaitez déployer le système sans serveur, AWS Apprunner est la voie à suivre.

Citation

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