py search images

CLI 도구 : 클립 및 벡터 임베딩을 사용한 이미지 검색

이 CLI 도구를 사용하면 지정된 폴더에서 이미지를 색인화하고 클립 (대비 언어 이미지 전 사전 레이싱), Pytorch 및 Numpy를 사용하여 텍스트 쿼리를 기반으로 유사한 이미지를 검색 할 수 있습니다. 이미지 벡터를 생성하고 시맨틱 검색을 효율적으로 수행 할 수 있습니다.

특징

• 이미지를 색인하고 벡터 임베딩을 생성합니다.
• 텍스트 쿼리를 기반으로 이미지를 검색하십시오.
• 여러 이미지 형식 (.jpg, .jpeg, .png, .webp)을 지원합니다.

용법

텍스트 쿼리로 이미지를 검색하려면 먼저이 도구를 사용하여 이미지가 포함 된 폴더를 색인해야합니다.

색인 이미지

폴더 images 에서 이미지를 색인하려면 실행합니다.

(venv) ➜ vector-embeddings python main.py index -f images

이 명령은 이미지 폴더의 이미지를 처리하고 클립 모델을 사용하여 벡터 임베딩을 생성 한 다음 동일한 폴더의 벡터.npz에 저장합니다.

관련 이미지를 검색하십시오

텍스트 쿼리를 기반으로 유사한 이미지를 검색하려면 실행하십시오.

(venv) ➜ vector-embeddings python main.py search " Your text query " -f images

"텍스트 쿼리"를 원하는 이미지에 대한 설명으로 바꾸십시오. 이 도구는 벡터에서 벡터를로드하고 가장 유사한 이미지를 반환합니다.