做了一个向量相乘的APP,避免embedding_mobile递归循环.

toAndroid/toAndroidEmbedding.py

@@ -15,7 +15,8 @@ def main():
     print("开始转换ResNet50 Embedding模型为移动端格式...")
     
     # 1. 加载训练好的embedding模型权重
-    base_model = create_resnet50_embedding(embedding_dim=512, pretrained=True)
+    # base_model = create_resnet50_embedding(embedding_dim=512, pretrained=True)
+    base_model = create_mobile_resnet50_embedding(embedding_dim=512, pretrained=True)
     model_path = "../model/embedding_20250917_145342/best_embedding_model.pth"
     
     if not os.path.exists(model_path):

toAndroid/toAndroidSearch.py

@@ -127,11 +127,11 @@ class TwoLayerSimilarityNet(nn.Module):
             raise ValueError("输入张量形状必须为 [batch, 512]")
 
         # 输入归一化
-        # x_norm = F.normalize(x, p=2, dim=1)  # [B, 512]
+        x_norm = F.normalize(x, p=2, dim=1)  # [B, 512]
 
         # 单层线性变换
-        # sims = self.layer2(x_norm)          # [B, N]
-        sims = self.layer2(x)          # [B, N]
+        sims = self.layer2(x_norm)          # [B, N]
+        # sims = self.layer2(x)          # [B, N]
 
         return sims