sis

간단한 이미지 검색 엔진

데모

워크 플로

소식

• [2020.06] CVPR 2020 튜토리얼 용 코드의 많은 부분을 업데이트했습니다.

개요

• Keras + Flask를 사용한 간단한 이미지 기반 이미지 검색 엔진. 두 개의 파이썬 스크립트를 실행하여 검색 엔진을 시작할 수 있습니다.
• offline.py :이 스크립트는 각 데이터베이스 이미지에서 깊은 기능을 추출합니다. 각 기능은 ImageNet 미리 훈련 된 가중치가있는 VGG16 모델의 4096D FC6 활성화입니다.
• server.py :이 스크립트는 웹 서버로 실행됩니다. 플라스크 웹 인터페이스를 통해 쿼리 이미지를 서버로 보낼 수 있습니다. 서버는 간단한 선형 스캔으로 쿼리와 유사한 이미지를 찾습니다.
• 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.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)로 검색 엔진을 실행하는 것이 좋습니다.
• (고급) 시스템을 서버로 배포하려면 AWS 승인자가 갈 길입니다.

소환

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