调整模型路径。
This commit is contained in:
@@ -1135,6 +1135,7 @@ class EmbeddingFoodClassifierApp:
|
||||
|
||||
# 提取查询图片的特征向量
|
||||
query_embedding = self.model.extract_embedding(pil_image, normalize=True)
|
||||
# print('特征向量:', query_embedding)
|
||||
query_embedding = query_embedding.reshape(1, -1).astype(np.float32)
|
||||
|
||||
# 在FAISS索引中搜索最相似的k张图片
|
||||
|
||||
Reference in New Issue
Block a user