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

关于保存特征的一些代码

Browse Source
This commit is contained in:
zchen
2024-03-26 18:00:35 +08:00
parent 7dce70027a
commit 88c815dd9d
16 changed files with 378 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/api/api_outfit_matcher.py
View File

@@ -27,7 +27,9 @@ def outfit_matcher(request_item: OutfitMatcher):
start_time = time.time()
for item in request_item['query']:
outfits = fashion_dataset.generate_outfit(item, request_item["topk"], request_item["max_outfits"])
scores = service.get_result(outfits)
scores, features = service.get_result(outfits)
# save features in databases
if request_item['is_best']:
best_outfits, best_scores = service.visualize(outfits, scores, request_item["topk"], best=True,
# output_path=os.path.join(r"E:\workspace\outfit_matcher\2024 SS Outfit", f"{item['item_name']}_best_{param['topk']}.png")