This commit is contained in:
pangkaicheng
2024-03-28 10:38:16 +08:00
110 changed files with 1418 additions and 1833 deletions

View File

@@ -30,7 +30,18 @@ if __name__ == '__main__':
prepared_feature[item["item_name"]] = np.load(f'feature/{item["item_name"]}.npy')
for item in tqdm(param["query"] * 10):
outfits = fashion_dataset.generate_outfit(item, param["topk"], param["max_outfits"])
<<<<<<< HEAD
scores = service.get_result(outfits, prepared_feature)
=======
scores, features = service.get_result(outfits)
# save features
# 链接milvus
# 存入数据库
# 关闭链接
>>>>>>> 1f23781b16e59bfbcbbb4d252e6a61685267e6c7
# print(scores)
# print(len(scores))
# service.visualize(outfits, scores, param["topk"], best=True,