sis

シンプルな画像検索エンジン

デモ

ワークフロー

ニュース

• [2020.06] CVPR 2020チュートリアルのコードの多くの部分を更新しました

概要

• Keras + Flaskを使用したシンプルな画像ベースの画像検索エンジン。 2つのPythonスクリプトを実行するだけで、検索エンジンを起動できます。
• offline.py ：このスクリプトは、各データベースイメージから深い機能を抽出します。各機能は、ImagENET事前訓練を受けた重みを備えたVGG16モデルからの4096D FC6アクティベーションです。
• server.py ：このスクリプトはWebサーバーを実行します。 Flask Web-Interfaceを介してクエリ画像をサーバーに送信できます。サーバーは、単純な線形スキャンによってクエリと同様の画像を見つけます。
• GPUは必要ありません。
• Ubuntu 18.04およびWSL2（Ubuntu 20.04）でテスト

リンク

• デモ
• CVPR2020 [スライド] [ビデオ]でのコース
• プロジェクトページ
• @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が開かれるようにセキュリティグループを作成する必要があることに注意してください。
• 中間レベルのCPUインスタンスで十分です。たとえば、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 :5000のようなものを介してブラウザからサーバーにアクセスできます。
• （Advanced）安全な方法でシステムを展開したい場合は、通常のWebサーバー（UWSGI + nginx）で検索エンジンを実行することを検討してください。
• （Advanced）システムをサーバーに展開する場合は、AWS Apprunnerが進む方法です。

引用

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