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

属性检测接口，响应字段规范

Browse Source
This commit is contained in:
zhouchengrong
2024-03-27 16:26:34 +08:00
parent 1f1553eb54
commit 864ce06439
102 changed files with 2325 additions and 1823 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
app/api/api_outfit_matcher.py
View File

@@ -20,27 +20,28 @@ def outfit_matcher(request_item: OutfitMatcher):
for i in range(len(request_item['database'])):
request_item['database'][i] = dict(request_item['database'][i])
try:
fashion_dataset = FashionDataset(request_item['database'])
service = OutfitMaterTypeAware()
result = []
start_time = time.time()
for item in request_item['query']:
outfits = fashion_dataset.generate_outfit(item, request_item["topk"], request_item["max_outfits"])
scores, features = service.get_result(outfits)
# save features in databases
# try:
fashion_dataset = FashionDataset(request_item['database'])
service = OutfitMaterTypeAware()
result = []
start_time = time.time()
for item in request_item['query']:
outfits = fashion_dataset.generate_outfit(item, request_item["topk"], request_item["max_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")
)
result.append({"outfits": best_outfits, "scores": best_scores})
else:
bad_outfits, bad_scores = service.visualize(outfits, scores, request_item["topk"], best=False,
# output_path=os.path.join(r"E:\workspace\outfit_matcher\2024 SS Outfit", f"{item['item_name']}_worst_{param['topk']}.png")
)
result.append({"outfits": bad_outfits, "scores": bad_scores})
logger.info(f"run time is : {time.time() - start_time}")
return {"message": "ok", "data": result}
except Exception as e:
return {"message": f"{e}", "data": e}
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")
)
result.append({"outfits": best_outfits, "scores": best_scores})
else:
bad_outfits, bad_scores = service.visualize(outfits, scores, request_item["topk"], best=False,
# output_path=os.path.join(r"E:\workspace\outfit_matcher\2024 SS Outfit", f"{item['item_name']}_worst_{param['topk']}.png")
)
result.append({"outfits": bad_outfits, "scores": bad_scores})
logger.info(f"run time is : {time.time() - start_time}")
return {"message": "ok", "data": result}
# except Exception as e:
# logger.warning(e)
# return {"message": f"{e}", "data": e}