确保图片压缩都采用F.interpolate(),这样和安卓端得到的结果基本差不多。

classifier/embedding_food_classifier_search_app.py

@@ -1124,7 +1124,7 @@ class EmbeddingFoodClassifierApp:
             query_tensor = torch.from_numpy(query_embedding.reshape(1, -1))  # [1,512]
             # 打印query_tensor的前十个数字
 
-            print(query_embedding[:10])
+            # print(query_embedding[:50])
 
             if self.db_matrix is None:
                 raise RuntimeError("向量库未加载")
@@ -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 = []