diff --git a/app/service/attribute_recognition/config/category_mapping.py b/app/service/attribute_recognition/config/category_mapping.py index f98546b..9951546 100644 --- a/app/service/attribute_recognition/config/category_mapping.py +++ b/app/service/attribute_recognition/config/category_mapping.py @@ -12,7 +12,7 @@ category_mapping = { "ACCESSORY/BIB": "accessories", "ACCESSORY/GIFT": "accessories", "ACCESSORY/FESTIVAL SET": "accessories", - "ACCESSORY/CAPE": "outerwear", + "ACCESSORY/CAPE": "outwear", "ACCESSORY/BLANKET": "accessories", "ACCESSORY/HAIR BAND": "accessories", "ACCESSORY/GLOVES": "accessories", @@ -44,8 +44,8 @@ category_mapping = { "BOTTOM/BOTTOM": "bottoms", "BOTTOM/CAPRI": "bottoms", "BOTTOM/": "bottoms", - "BOTTOM/COAT": "outerwear", - "BOTTOM/CARDIGAN": "outerwear", + "BOTTOM/COAT": "outwear", + "BOTTOM/CARDIGAN": "outwear", "BOTTOM/JEANS SHORTS": "bottoms", "BOTTOM/BOOTS": "shoes", "BOTTOM/JEANS": "bottoms", @@ -53,8 +53,8 @@ category_mapping = { "BOTTOM/JUMPSUIT": "all-body", "BOTTOM/JEANS SKIRT": "skirt", "BOTTOM/JEANS CARPI": "bottoms", - "BOTTOM/JACKET": "outerwear", - "BOTTOM/LEATHER JACKET": "outerwear", + "BOTTOM/JACKET": "outwear", + "BOTTOM/LEATHER JACKET": "outwear", "BOTTOM/LEATHER PANTS": "bottoms", "BOTTOM/JEANS PANTS": "bottoms", "BOTTOM/PANTS": "bottoms", @@ -71,7 +71,7 @@ category_mapping = { "BOTTOM/SHOES": "shoes", "BOTTOM/SPORTSWEAR PANTS": "bottoms", "BOTTON/PANTS": "bottoms", - "DRESS/JEANS JACKET": "outerwear", + "DRESS/JEANS JACKET": "outwear", "DRESS/": "all-body", "DRESS/ONE PIECE": "all-body", "DRESS/DRESS": "all-body", @@ -87,75 +87,75 @@ category_mapping = { "DRESS/VEST": "tops", "DRESS/SHIRT": "tops", "DRESS/TUTA": "all-body", - "FUR/DOWN COAT": "outerwear", + "FUR/DOWN COAT": "outwear", "FUR/HAT": "hats", - "FUR/FUR CAPE": "outerwear", - "FUR/FUR COAT": "outerwear", - "FUR/FUR DOWN JACKET": "outerwear", - "FUR/FUR GILET": "outerwear", - "FUR/FUR CARDIGAN": "outerwear", - "FUR/FUR JACKET": "outerwear", + "FUR/FUR CAPE": "outwear", + "FUR/FUR COAT": "outwear", + "FUR/FUR DOWN JACKET": "outwear", + "FUR/FUR GILET": "outwear", + "FUR/FUR CARDIGAN": "outwear", + "FUR/FUR JACKET": "outwear", "FUR/FUR HAT": "hats", "FUR/FUR SCARF": "accessories", - "FUR/FUR VEST": "outerwear", + "FUR/FUR VEST": "outwear", "GIFT/GIFT": "accessories", "GIFT/SCARF": "accessories", - "JEANS/GILET": "outerwear", + "JEANS/GILET": "outwear", "JEANS/JEANS CARPI": "bottoms", "JEANS/DRESS": "all-body", "JEANS/JEANS BLOUSE": "tops", "JEANS/JEANS": "bottoms", "JEANS/JEANS OVERALL": "all-body", - "JEANS/JEANS BLAZER": "outerwear", + "JEANS/JEANS BLAZER": "outwear", "JEANS/JEANS DRESS": "all-body", "JEANS/SHIRT": "tops", "JEANS/JEANS SHORTS": "bottoms", - "JEANS/JEANS GILET": "outerwear", - "JEANS/JEANS JACKET": "outerwear", + "JEANS/JEANS GILET": "outwear", + "JEANS/JEANS JACKET": "outwear", "JEANS/SKIRT": "skirt", "JEANS/TWIN SET": "all-body", "JEANS/JEANS PANTS": "bottoms", "JEANS/JEANS SHIRT": "tops", "JEANS/JEANS SKIRT": "skirt", - "JEANS/JEANS VEST": "outerwear", + "JEANS/JEANS VEST": "outwear", "JEANS/JEANS TOP": "tops", "JEANS/OVERALL": "all-body", "JEANS/PANTS": "bottoms", "JEANS/SHORTS": "bottoms", - "KNIT/CARDGAIN": "outerwear", + "KNIT/CARDGAIN": "outwear", "KNIT/ONE PIECE": "all-body", - "KNIT/CARDIGAN": "outerwear", + "KNIT/CARDIGAN": "outwear", "KNIT/BRA TOP": "tops", - "KNIT/JACKET": "outerwear", + "KNIT/JACKET": "outwear", "KNIT/PULLOVER": "tops", "KNIT/DRESS": "all-body", - "KNIT/CAPE": "outerwear", + "KNIT/CAPE": "outwear", "KNIT/KNIT TOP": "tops", - "KNIT/CARDGIAN": "outerwear", + "KNIT/CARDGIAN": "outwear", "KNIT/PANTS": "bottoms", - "KNIT/GILET": "outerwear", + "KNIT/GILET": "outwear", "KNIT/LONG TOP": "tops", "KNIT/SKIRT": "skirt", "KNIT/SHORTS": "bottoms", "KNIT/SWEATER": "tops", "KNIT/SWEATPR": "tops", - "KNIT/VEST": "outerwear", + "KNIT/VEST": "outwear", "KNIT/TANK": "tops", "KNIT/TOP": "tops", "KNIT/TUTA": "all-body", "KNIT/TWIN SET": "all-body", - "LEATHER/BLAZER": "outerwear", - "LEATHER/LEATHER BLAZER": "outerwear", - "LEATHER/LEATHER COAT": "outerwear", + "LEATHER/BLAZER": "outwear", + "LEATHER/LEATHER BLAZER": "outwear", + "LEATHER/LEATHER COAT": "outwear", "LEATHER/LEATHER PANTS": "bottoms", "LEATHER/HAT": "hats", "LEATHER/LEATHER BLOUSE": "tops", "LEATHER/LEATHER DRESS": "all-body", "LEATHER/LEATHER SHORTS": "bottoms", "LEATHER/LEATHER SKIRT": "skirt", - "LEATHER/LEATHER GILET": "outerwear", - "LEATHER/LEATHER JACKET": "outerwear", - "LEATHER/LEATHER VEST": "outerwear", + "LEATHER/LEATHER GILET": "outwear", + "LEATHER/LEATHER JACKET": "outwear", + "LEATHER/LEATHER VEST": "outwear", "LEATHER/LEATHERSKIRT": "skirt", "LEATHER/LEATHERSHIRT": "tops", "LEATHER/LEATHER TOP": "tops", @@ -167,57 +167,57 @@ category_mapping = { "ONE PIECE/OVERALL": "all-body", "ONE PIECE/SWIMSUIT": "all-body", "ONE PIECE/TWIN SET": "all-body", - "OUTERWEAR/CARDIGAN": "outerwear", - "OUTERWEAR/BLAZER": "outerwear", - "OUTERWEAR/DOWN COAT": "outerwear", - "OUTERWEAR/": "outerwear", - "OUTERWEAR/DOWN JACKET": "outerwear", - "OUTERWEAR/COAT": "outerwear", - "OUTERWEAR/DOWN GILET": "outerwear", - "OUTERWEAR/CAPE": "outerwear", - "OUTERWEAR/DOWN VEST": "outerwear", - "OUTERWEAR/FUR JACKET": "outerwear", - "OUTERWEAR/FUR VEST": "outerwear", - "OUTERWEAR/CAPE COAT": "outerwear", - "OUTERWEAR/LEATHER VEST": "outerwear", - "OUTERWEAR/PULLOVER": "outerwear", - "OUTERWEAR/GILET": "outerwear", - "OUTERWEAR/FUR COAT": "outerwear", - "OUTERWEAR/QUILTED JACKET": "outerwear", - "OUTERWEAR/SPORTSWEAR JACKET": "outerwear", - "OUTERWEAR/JACKET": "outerwear", - "OUTERWEAR/FUR GILET": "outerwear", - "OUTERWEAR/SHIRT": "tops", - "OUTERWEAR/TWIN SET": "all-body", - "OUTERWEAR/LEATHER GILET": "outerwear", - "OUTERWEAR/JEANS JACKET": "outerwear", - "OUTERWEAR/TRENCH COAT": "outerwear", - "OUTERWEAR/LEATHER COAT": "outerwear", - "OUTERWEAR/ONE PIECE": "all-body", - "OUTERWEAR/LEATHER JACKET": "outerwear", - "OUTERWEAR/SCARF": "accessories", - "OUTERWEAR/SOCKS": "accessories", - "OUTERWEAR/TOP": "tops", - "OUTERWEAR/UNIFORM": "outerwear", - "OUTERWEAR/VEST": "outerwear", - "OUTERWEAR/WINDBREAKER": "outerwear", - "OUTERWEAR/WINDERBREAKER": "outerwear", - "OUTPRWEAR/BLAZER": "outerwear", + "outwear/CARDIGAN": "outwear", + "outwear/BLAZER": "outwear", + "outwear/DOWN COAT": "outwear", + "outwear/": "outwear", + "outwear/DOWN JACKET": "outwear", + "outwear/COAT": "outwear", + "outwear/DOWN GILET": "outwear", + "outwear/CAPE": "outwear", + "outwear/DOWN VEST": "outwear", + "outwear/FUR JACKET": "outwear", + "outwear/FUR VEST": "outwear", + "outwear/CAPE COAT": "outwear", + "outwear/LEATHER VEST": "outwear", + "outwear/PULLOVER": "outwear", + "outwear/GILET": "outwear", + "outwear/FUR COAT": "outwear", + "outwear/QUILTED JACKET": "outwear", + "outwear/SPORTSWEAR JACKET": "outwear", + "outwear/JACKET": "outwear", + "outwear/FUR GILET": "outwear", + "outwear/SHIRT": "tops", + "outwear/TWIN SET": "all-body", + "outwear/LEATHER GILET": "outwear", + "outwear/JEANS JACKET": "outwear", + "outwear/TRENCH COAT": "outwear", + "outwear/LEATHER COAT": "outwear", + "outwear/ONE PIECE": "all-body", + "outwear/LEATHER JACKET": "outwear", + "outwear/SCARF": "accessories", + "outwear/SOCKS": "accessories", + "outwear/TOP": "tops", + "outwear/UNIFORM": "outwear", + "outwear/VEST": "outwear", + "outwear/WINDBREAKER": "outwear", + "outwear/WINDERBREAKER": "outwear", + "OUTPRWEAR/BLAZER": "outwear", "SHOES/PUMPS": "shoes", "SHOES/BOOTS": "shoes", "SHOES/SHOES": "shoes", "SHOES/SANDALS": "shoes", "TOP/BLOUSE": "tops", - "TOP/BLAZER": "outerwear", + "TOP/BLAZER": "outwear", "TOP/BAR TOP": "tops", "TOP/BRA TOP": "tops", - "TOP/CARDIGAN": "outerwear", + "TOP/CARDIGAN": "outwear", "TOP/BODY": "all-body", - "TOP/COAT": "outerwear", + "TOP/COAT": "outwear", "TOP/PANTS": "bottoms", - "TOP/DOWN VEST": "outerwear", - "TOP/JACKET": "outerwear", - "TOP/GILET": "outerwear", + "TOP/DOWN VEST": "outwear", + "TOP/JACKET": "outwear", + "TOP/GILET": "outwear", "TOP/TWIN SET": "tops", "TOP/LONG SLEEVES TEE": "tops", "TOP/LONG TOP": "tops", @@ -225,7 +225,7 @@ category_mapping = { "TOP/VEST": "tops", "TOP/ONE PIECE": "all-body", "TOP/TANK": "tops", - "TOP/PULLOVER": "outerwear", + "TOP/PULLOVER": "outwear", "TOP/TOP": "tops", "TOP/OVERALL": "all-body", "TOP/UNDERWEAR": "bottoms", diff --git a/app/service/attribute_recognition/service.py b/app/service/attribute_recognition/service.py index 96112b1..2044086 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 == 'outerwear': + elif category == 'outwear': attr_dict = {'Item': 'outer'} for i in range(len(args.outwear_discription_list)):