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

2025-10-09 17:58:06 +08:00
parent 065192fff8
commit 7faf3f6480
4 changed files with 1554 additions and 20 deletions
@@ -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):