diff --git a/app/service/attribute_recognition/service.py b/app/service/attribute_recognition/service.py index 2044086..96112b1 100644 --- a/app/service/attribute_recognition/service.py +++ b/app/service/attribute_recognition/service.py @@ -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 == 'outerwear': attr_dict = {'Item': 'outer'} for i in range(len(args.outwear_discription_list)):