OpenCV faiss를 이용한 유사 이미지 검색 프로그램


OpenCV faiss를 이용한 유사 이미지 검색 프로그램

이미지의 특징점을 추출하여 faiss를 이용해 계산해서 입력이미지와 유사한 이미지를 찾는 프로그램, 특징점의 오차가 0에 가까울 수록 비슷하고 멀 수록 다르다. 코드 import cv2 import numpy as np import faiss # vlad 계산 def compute_vlad(image, des, labels, centers): sift = cv2.SIFT_create(edgeThreshold=80) _, des = sift.detectAndCompute(image, None) if des is None: return None vlad = np.zeros((centers.shape[0], des.shape[1]), dtype=np.float32) for i in range(des.shape[0]): vlad[labels[i]] += des[i] - centers[labels[i]] vlad = cv2.normalize(vlad, None).flatten() vlad /...



원문링크 : OpenCV faiss를 이용한 유사 이미지 검색 프로그램