From 3417bcb2ab1ba4858d5f7db41ddc59413be04ce0 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Fri, 30 Aug 2024 12:36:56 +0800 Subject: [PATCH] =?UTF-8?q?fix=20=E7=B1=BB=E5=88=AB=E6=98=A0=E5=B0=84?= =?UTF-8?q?=E8=A1=A8=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/attribute_recognition/service.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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)):