aidlab/sora_python
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Picture using minio service

Browse Source
This commit is contained in:
zhouchengrong
2024-03-18 14:17:43 +08:00
parent 26c328b1e4
commit 2f89e7432f
116 changed files with 10294 additions and 254 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
app/service/outfit_matcher/service.py
View File

@@ -1,25 +1,30 @@
import json
import os
from pprint import pprint
from app.service.outfit_matcher.dataset import FashionDataset
from app.service.outfit_matcher.outfit_evaluator import OutfitMaterTypeAware
if __name__ == '__main__':
with open("./test_param/recommendation_test.json", "r") as f:
param = json.load(f)
fashion_dataset = FashionDataset(param["database"])
service = OutfitMaterTypeAware()
best_list = []
bad_list = []
for item in param["query"]:
outfits = fashion_dataset.generate_outfit(item, param["topk"], param["max_outfits"])
scores = service.get_result(outfits)
print(scores)
# service.visualize(outfits, scores, param["topk"], best=True,
# output_path=os.path.join(
# r"E:\workspace\outfit_matcher\2024 SS Outfit",
# f"{item['item_name']}_best_{param['topk']}.png"
# ))
# service.visualize(outfits, scores, param["topk"], best=False,
# output_path=os.path.join(
# r"E:\workspace\outfit_matcher\2024 SS Outfit",
# f"{item['item_name']}_worst_{param['topk']}.png"
# ))
# print(scores)
# print(len(scores))
best_outfits, best_scores = service.visualize(outfits, scores, param["topk"], best=True,
# output_path=os.path.join(r"E:\workspace\outfit_matcher\2024 SS Outfit", f"{item['item_name']}_best_{param['topk']}.png")
)
bad_outfits, bad_scores = service.visualize(outfits, scores, param["topk"], best=False,
# output_path=os.path.join(r"E:\workspace\outfit_matcher\2024 SS Outfit", f"{item['item_name']}_worst_{param['topk']}.png")
)
best_list.append({"best_outfits": best_outfits, "best_scores": best_scores})
bad_list.append({"bad_outfits": bad_outfits, "bad_scores": bad_scores})
pprint(best_list)
pprint(bad_list)