From 145b0d0f8cfe2c9333f32e07cddf3e6b42fbdcc3 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Thu, 28 Mar 2024 13:27:00 +0800 Subject: [PATCH] =?UTF-8?q?attribute=20=E5=AD=97=E6=AE=B5=E5=90=8D?= =?UTF-8?q?=E8=A7=84=E8=8C=83?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/attribute_recognition/service.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/app/service/attribute_recognition/service.py b/app/service/attribute_recognition/service.py index 5ca5927..df6e295 100644 --- a/app/service/attribute_recognition/service.py +++ b/app/service/attribute_recognition/service.py @@ -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)):