sis

簡單的圖像搜索引擎

演示

工作流程

消息

• [2020.06]更新了CVPR 2020教程的代碼的許多部分

概述

• 簡單的基於圖像的圖像搜索引擎使用KERAS +燒瓶。您只需運行兩個Python腳本即可啟動搜索引擎。
• offline.py ：此腳本從每個數據庫圖像中提取深度功能。每個功能都是來自具有Imagenet預訓練重量的VGG16模型的4096D FC6激活。
• server.py ：此腳本運行一個網絡服務器。您可以通過燒瓶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

高級：在AWS上啟動

• 您可以輕鬆地在AWS EC2上啟動搜索引擎服務器。請首先打開端口5000並啟動EC2實例。請注意，您需要創建一個安全組，以便打開端口5000。
• 中級CPU實例就足夠了，例如，m5。
• 通過SSH登錄實例後，請設置Python環境（例如，Anaconda）。
• offline.py and server.py運行。
• 運行python server.py後，您可以通過http://ec2-XX-XX-XXX-XXX.us-west-2.compute.amazonaws.com:5000 from瀏覽器訪問服務器。
• （高級）如果您想以安全的方式部署系統，請考慮使用通常的Web服務器（例如，UWSGI + NGINX）運行搜索引擎。
• （高級）如果您想無服務器部署系統，那麼AWS評估儀是必經之路。

引用

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