From a4f028c25df102c3eb975f1769e8aca613d03c5f Mon Sep 17 00:00:00 2001 From: ldy Date: Fri, 13 Jun 2025 07:16:28 +0000 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20find=5Fsimilar=5Fimg.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- 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 ———