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

修复 dataset outerwear出现的key error问题

Browse Source
This commit is contained in:
zhouchengrong
2024-03-20 17:21:59 +08:00
parent c536e1ad36
commit 7dce70027a
2 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
app/service/outfit_matcher/dataset.py
View File

@@ -80,12 +80,12 @@ class FashionDataset(object):
used_items.add(item["item_name"]) used_items.add(item["item_name"])
outfit = [query_item, item] outfit = [query_item, item]
outfit_list.append(tuple(outfit)) outfit_list.append(tuple(outfit))
if "outerwear" in self.cate2item.keys():
# 20% chance to include an outerwear # 20% chance to include an outerwear
if self.cate2num['outerwear'] > 0 and random.random() < 0.2: if self.cate2num['outerwear'] > 0 and random.random() < 0.2:
outerwear = random.choice(self.cate2item['outerwear']) outerwear = random.choice(self.cate2item['outerwear'])
outfit.append(outerwear) outfit.append(outerwear)
outfit_list.append(tuple(outfit)) outfit_list.append(tuple(outfit))
if len(outfit_list) < topk: if len(outfit_list) < topk:
raise ValueError(f"Cannot generate more than {topk} outfits!") raise ValueError(f"Cannot generate more than {topk} outfits!")

4
app/service/outfit_matcher/test_param/recommendation_test_zcr.json
View File

@@ -1,6 +1,6 @@
{ {
"topk": 5, "topk": 1,
"max_outfits": 100, "max_outfits": 10,
"is_best": true, "is_best": true,
"query": [ "query": [
{ {