建立FAISS索引。

2025-09-23 15:34:23 +08:00
parent 2d5096a0b2
commit d51853997f
3 changed files with 524 additions and 2 deletions
@@ -153,7 +153,8 @@ class ResNet50EmbeddingNet(nn.Module):
        # 生成embedding
        embeddings = self._forward_embedding(backbone_features)
        
-        # L2归一化（用于余弦相似度计算）
+        # L2归一化（用于余弦相似度计算），p=2表示L2范数(即平方和再开根号)，dim=1表示在第二个维度上做归一化
+        # 形状是[batch_size, embedding_dim]
        if normalize:
            embeddings = F.normalize(embeddings, p=2, dim=1)