关于保存特征的一些代码

2024-03-26 18:00:35 +08:00
parent 7dce70027a
commit 88c815dd9d
16 changed files with 378 additions and 10 deletions
--- a/app/service/outfit_matcher/service.py
+++ b/app/service/outfit_matcher/service.py
@@ -14,7 +14,14 @@ if __name__ == '__main__':
    bad_list = []
    for item in param["query"]:
        outfits = fashion_dataset.generate_outfit(item, param["topk"], param["max_outfits"])
-        scores = service.get_result(outfits)
+        scores, features = service.get_result(outfits)
+        # save  features
+
+        # 链接milvus
+
+        # 存入数据库
+        # 关闭链接
+
        # print(scores)
        # print(len(scores))
        best_outfits, best_scores = service.visualize(outfits, scores, param["topk"], best=True,