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

attribute 字段名规范

Browse Source
This commit is contained in:
zhouchengrong
2024-03-28 13:27:00 +08:00
parent 75bedfbeac
commit 145b0d0f8c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -111,7 +111,7 @@ class AttributeRecognition:
attr_dict = {}
category = 'bottom'
attr_dict['Item'] = category
attr_dict['BTM_Type'] = ['Pants']
attr_dict['Type'] = ['Pants']
for i in range(len(args.bottom_discription_list)):
attr_description = args.bottom_discription_list[i]
attr_model_path = args.bottom_model_list[i]
@@ -119,7 +119,7 @@ class AttributeRecognition:
attr_dict = Merge(attr_dict, present_dict)
elif category == 'skirt':
attr_dict = {'BTM_Type': ['Skirt']}
attr_dict = {'Type': ['Skirt']}
category = 'bottom'
attr_dict['Item'] = category
for i in range(len(args.bottom_discription_list)):