Image-Retrieval

이 레포지토리는 Fly Ai 프로젝트 "ArtVisionXperience"에서 이미지 검색 기능을 구현한 것입니다.

간단한 이미지 검색 과정

1. 사전 훈련된 신경망(e.g. ResNet, VGG)을 로드합니다.
2. 가중치를 동결합니다.
3. 헤더(FC)를 제거하고 이미지의 피쳐를 벡터화한 평면 레이어만 남깁니다.
4. 모든 후보 이미지를 계산하고 벡터화된 이미지를 얻습니다.
5. 입력 이미지를 계산하고 벡터화된 이미지를 얻습니다.
6. 각 벡터화된 이미지 간의 코사인 유사도를 계산합니다.
7. 후보 이미지 중 가장 유사한 이미지를 선택합니다.