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

fix 类别映射表修改

Browse Source
This commit is contained in:
zhouchengrong
2024-08-30 12:36:56 +08:00
parent 450b0b2149
commit 3417bcb2ab
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
app/service/attribute_recognition/service.py
View File

@@ -102,7 +102,7 @@ class AttributeRecognition:
category = category_list[0]
print(category)
if category == 'tops':
if category == 'tops' or category == "top":
attr_dict = {'Item': "top"}
for i in range(len(args.top_discription_list)):
attr_description = args.top_discription_list[i]
@@ -110,7 +110,7 @@ class AttributeRecognition:
present_dict = self.get_attribute(attr_model_path, attr_description, sample)
attr_dict = Merge(attr_dict, present_dict)
elif category == 'bottoms':
elif category == 'bottoms' or category == 'bottom':
attr_dict = {'Item': "bottom", 'Type': ['Pants']}
for i in range(len(args.bottom_discription_list)):
attr_description = args.bottom_discription_list[i]
@@ -136,7 +136,7 @@ class AttributeRecognition:
present_dict = self.get_attribute(attr_model_path, attr_description, sample)
attr_dict = Merge(attr_dict, present_dict)
elif category == 'outwear':
elif category == 'outwear' or category == 'outerwear':
attr_dict = {'Item': 'outer'}
for i in range(len(args.outwear_discription_list)):