diff --git a/app/service/attribute_recognition/service.py b/app/service/attribute_recognition/service.py index 2044086..755e19b 100644 --- a/app/service/attribute_recognition/service.py +++ b/app/service/attribute_recognition/service.py @@ -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)):