ldy/Img2Vec
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新 find_similar_img.py

Browse Source
This commit is contained in:
ldy 2025-06-13 07:16:28 +00:00
parent 47b9279725
commit a4f028c25d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
find_similar_img.py
View File

@ -6,8 +6,8 @@ from sklearn.metrics.pairwise import cosine_similarity
from embedder import ImageEmbedder from embedder import ImageEmbedder
# ——— Load stored embeddings & mapping ——— # ——— Load stored embeddings & mapping ———
vecs = np.load("processed_images/image_vectors.npy") # (N, D) vecs = np.load("imgVecs/image_vectors.npy") # (N, D)
with open("processed_images/index_to_file.pkl", "rb") as f: with open("imgVecs/index_to_file.pkl", "rb") as f:
idx2file = pickle.load(f) # dict: idx → filepath idx2file = pickle.load(f) # dict: idx → filepath
# ——— Specify query image ——— # ——— Specify query image ———