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 17:09:24 +08:00
parent c8243c0f16
commit f12f2d1514
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/api/api_outfit_matcher.py
View File

@@ -32,12 +32,12 @@ def outfit_matcher(request_item: OutfitMatcher):
best_outfits, best_scores = service.visualize(outfits, scores, request_item["topk"], best=True, 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") # output_path=os.path.join(r"E:\workspace\outfit_matcher\2024 SS Outfit", f"{item['item_name']}_best_{param['topk']}.png")
) )
result.append({"best_outfits": best_outfits, "best_scores": best_scores}) result.append({"outfits": best_outfits, "scores": best_scores})
else: else:
bad_outfits, bad_scores = service.visualize(outfits, scores, request_item["topk"], best=False, 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") # output_path=os.path.join(r"E:\workspace\outfit_matcher\2024 SS Outfit", f"{item['item_name']}_worst_{param['topk']}.png")
) )
result.append({"bad_outfits": bad_outfits, "bad_scores": bad_scores}) result.append({"outfits": bad_outfits, "scores": bad_scores})
logger.info(f"run time is : {time.time() - start_time}") logger.info(f"run time is : {time.time() - start_time}")
return {"message": "ok", "data": result} return {"message": "ok", "data": result}
except Exception as e: except Exception as e: