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

.

Browse Source
This commit is contained in:
pangkaicheng
2024-03-12 12:13:46 +08:00
parent b2be4484a5
commit 26c328b1e4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -8,9 +8,9 @@ if __name__ == '__main__':
with open("./test_param/recommendation_test.json", "r") as f: with open("./test_param/recommendation_test.json", "r") as f:
param = json.load(f) param = json.load(f)
fashion_dataset = FashionDataset(param["database"]) fashion_dataset = FashionDataset(param["database"])
service = OutfitMaterTypeAware()
for item in param["query"]: for item in param["query"]:
outfits = fashion_dataset.generate_outfit(item, param["topk"], param["max_outfits"]) outfits = fashion_dataset.generate_outfit(item, param["topk"], param["max_outfits"])
service = OutfitMaterTypeAware()
scores = service.get_result(outfits) scores = service.get_result(outfits)
print(scores) print(scores)
# service.visualize(outfits, scores, param["topk"], best=True, # service.visualize(outfits, scores, param["topk"], best=True,