只是更改了一些配置

2025-10-10 15:59:01 +08:00
parent 9eed8acb71
commit 2660d503d6
4 changed files with 6 additions and 6 deletions
@@ -62,7 +62,7 @@ class EmbeddingFoodClassifierApp:
        """加载训练好的embedding模型和FAISS索引"""
        try:
            # 模型路径
-            model_path = "../model/embedding_20250930_102826/best_embedding_model.pth"
+            model_path = "../model/embedding_20251010_111257/best_embedding_model.pth"
            
            # FAISS索引目录
            index_dir = "../faiss_vector_db/faiss_index"
@@ -1136,8 +1136,8 @@ class EmbeddingFoodClassifierApp:
            topk_scores, topk_indices = torch.topk(sims, k=min(k, sims.shape[1]), dim=1)
            indices = topk_indices.cpu().numpy()
            scores = topk_scores.cpu().numpy()
-            # print("相似度索引：", indices)
-            # print("相似度分数：", scores)
+            print("相似度索引：", indices)
+            print("相似度分数：", scores)

            # 收集相似图片的类别
            similar_classes = []