Image to Image Search

SmartSearch는 캡션을 생성하고 해당 캡션을 비교하여 유사한 이미지를 찾는 리버스 이미지 검색 엔진입니다. Tensorflow 및 Elasticsearch로 구동됩니다.

업데이트 : Codeberg : Image-to-Image-Search 에서이 저장소의 미러

업데이트 : 이미지 캡션 생성기에 대한 REST API 및 이미지 검색이 추가됩니다! (xc0d3rz로 외침)

데모

딥 리버스 이미지 검색 엔진 - YouTube는 핵심 기능을 데모합니다.

가지

도커 이미지

기본 구현

필요한 패키지 :

• 아나콘다
• 텐서 플로우 백엔드가있는 케라 (Python 3.6)
• 탄성 검색 및 elasticsearch-py (탄성 검색 6.0)

자세한 내용은 요구 사항을 확인하십시오 .txt

미리 훈련 된 모델

• flickr-8k lstm weights (flickr8k_cnn_lstm_v1.p)

이것을 다운로드하고 모델 폴더 내부에 붙여 넣으십시오.

산출

팁

• Elasticsearch를 설치하고 Server.py 또는 Index_Database.py를 시작하기 전에 탄성 검색 프로세스가 실행 중인지 항상 확인하십시오.

• 업로드 기능을 사용하는 대신 static/img 폴더 내에 모든 이미지를 붙여 넣은 다음 python index_database.py 를 붙여서 모든 이미지를 색인화하십시오.

• 인덱스 된 이미지를 삭제하려면 sh delete_index.sh

감사해요

미학을위한 CSS를 실현 덕분에 플라스크 튜토리얼 용으로 인쇄하고 멋진 캡션 라이브러리를위한 Andrej Karpathy를 인쇄합니다. REST API 포함을 위해 XC0D3RZ에게 외침!