From 70b9a14b2ba099c24ac756f8879f87e58da5eb43 Mon Sep 17 00:00:00 2001 From: ldy Date: Fri, 13 Jun 2025 15:15:24 +0800 Subject: [PATCH] Updated deprecated path --- find_similar_img.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/find_similar_img.py b/find_similar_img.py index cc8e8f5..6df859e 100644 --- a/find_similar_img.py +++ b/find_similar_img.py @@ -6,8 +6,8 @@ from sklearn.metrics.pairwise import cosine_similarity from embedder import ImageEmbedder # ——— Load stored embeddings & mapping ——— -vecs = np.load("processed_images/image_vectors.npy") # (N, D) -with open("processed_images/index_to_file.pkl", "rb") as f: +vecs = np.load("imgVecs/image_vectors.npy") # (N, D) +with open("imgVecs/index_to_file.pkl", "rb") as f: idx2file = pickle.load(f) # dict: idx → filepath # ——— Specify query image ———