aidlab/sora_python
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix 类别映射表修改

Browse Source
This commit is contained in:
zhouchengrong
2024-08-30 11:01:07 +08:00
parent a31952ba0d
commit 766ab91b68
2 changed files with 76 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

150
app/service/attribute_recognition/config/category_mapping.py
View File

@@ -12,7 +12,7 @@ category_mapping = {
"ACCESSORY/BIB": "accessories", "ACCESSORY/BIB": "accessories",
"ACCESSORY/GIFT": "accessories", "ACCESSORY/GIFT": "accessories",
"ACCESSORY/FESTIVAL SET": "accessories", "ACCESSORY/FESTIVAL SET": "accessories",
"ACCESSORY/CAPE": "outerwear", "ACCESSORY/CAPE": "outwear",
"ACCESSORY/BLANKET": "accessories", "ACCESSORY/BLANKET": "accessories",
"ACCESSORY/HAIR BAND": "accessories", "ACCESSORY/HAIR BAND": "accessories",
"ACCESSORY/GLOVES": "accessories", "ACCESSORY/GLOVES": "accessories",
@@ -44,8 +44,8 @@ category_mapping = {
"BOTTOM/BOTTOM": "bottoms", "BOTTOM/BOTTOM": "bottoms",
"BOTTOM/CAPRI": "bottoms", "BOTTOM/CAPRI": "bottoms",
"BOTTOM/": "bottoms", "BOTTOM/": "bottoms",
"BOTTOM/COAT": "outerwear", "BOTTOM/COAT": "outwear",
"BOTTOM/CARDIGAN": "outerwear", "BOTTOM/CARDIGAN": "outwear",
"BOTTOM/JEANS SHORTS": "bottoms", "BOTTOM/JEANS SHORTS": "bottoms",
"BOTTOM/BOOTS": "shoes", "BOTTOM/BOOTS": "shoes",
"BOTTOM/JEANS": "bottoms", "BOTTOM/JEANS": "bottoms",
@@ -53,8 +53,8 @@ category_mapping = {
"BOTTOM/JUMPSUIT": "all-body", "BOTTOM/JUMPSUIT": "all-body",
"BOTTOM/JEANS SKIRT": "skirt", "BOTTOM/JEANS SKIRT": "skirt",
"BOTTOM/JEANS CARPI": "bottoms", "BOTTOM/JEANS CARPI": "bottoms",
"BOTTOM/JACKET": "outerwear", "BOTTOM/JACKET": "outwear",
"BOTTOM/LEATHER JACKET": "outerwear", "BOTTOM/LEATHER JACKET": "outwear",
"BOTTOM/LEATHER PANTS": "bottoms", "BOTTOM/LEATHER PANTS": "bottoms",
"BOTTOM/JEANS PANTS": "bottoms", "BOTTOM/JEANS PANTS": "bottoms",
"BOTTOM/PANTS": "bottoms", "BOTTOM/PANTS": "bottoms",
@@ -71,7 +71,7 @@ category_mapping = {
"BOTTOM/SHOES": "shoes", "BOTTOM/SHOES": "shoes",
"BOTTOM/SPORTSWEAR PANTS": "bottoms", "BOTTOM/SPORTSWEAR PANTS": "bottoms",
"BOTTON/PANTS": "bottoms", "BOTTON/PANTS": "bottoms",
"DRESS/JEANS JACKET": "outerwear", "DRESS/JEANS JACKET": "outwear",
"DRESS/": "all-body", "DRESS/": "all-body",
"DRESS/ONE PIECE": "all-body", "DRESS/ONE PIECE": "all-body",
"DRESS/DRESS": "all-body", "DRESS/DRESS": "all-body",
@@ -87,75 +87,75 @@ category_mapping = {
"DRESS/VEST": "tops", "DRESS/VEST": "tops",
"DRESS/SHIRT": "tops", "DRESS/SHIRT": "tops",
"DRESS/TUTA": "all-body", "DRESS/TUTA": "all-body",
"FUR/DOWN COAT": "outerwear", "FUR/DOWN COAT": "outwear",
"FUR/HAT": "hats", "FUR/HAT": "hats",
"FUR/FUR CAPE": "outerwear", "FUR/FUR CAPE": "outwear",
"FUR/FUR COAT": "outerwear", "FUR/FUR COAT": "outwear",
"FUR/FUR DOWN JACKET": "outerwear", "FUR/FUR DOWN JACKET": "outwear",
"FUR/FUR GILET": "outerwear", "FUR/FUR GILET": "outwear",
"FUR/FUR CARDIGAN": "outerwear", "FUR/FUR CARDIGAN": "outwear",
"FUR/FUR JACKET": "outerwear", "FUR/FUR JACKET": "outwear",
"FUR/FUR HAT": "hats", "FUR/FUR HAT": "hats",
"FUR/FUR SCARF": "accessories", "FUR/FUR SCARF": "accessories",
"FUR/FUR VEST": "outerwear", "FUR/FUR VEST": "outwear",
"GIFT/GIFT": "accessories", "GIFT/GIFT": "accessories",
"GIFT/SCARF": "accessories", "GIFT/SCARF": "accessories",
"JEANS/GILET": "outerwear", "JEANS/GILET": "outwear",
"JEANS/JEANS CARPI": "bottoms", "JEANS/JEANS CARPI": "bottoms",
"JEANS/DRESS": "all-body", "JEANS/DRESS": "all-body",
"JEANS/JEANS BLOUSE": "tops", "JEANS/JEANS BLOUSE": "tops",
"JEANS/JEANS": "bottoms", "JEANS/JEANS": "bottoms",
"JEANS/JEANS OVERALL": "all-body", "JEANS/JEANS OVERALL": "all-body",
"JEANS/JEANS BLAZER": "outerwear", "JEANS/JEANS BLAZER": "outwear",
"JEANS/JEANS DRESS": "all-body", "JEANS/JEANS DRESS": "all-body",
"JEANS/SHIRT": "tops", "JEANS/SHIRT": "tops",
"JEANS/JEANS SHORTS": "bottoms", "JEANS/JEANS SHORTS": "bottoms",
"JEANS/JEANS GILET": "outerwear", "JEANS/JEANS GILET": "outwear",
"JEANS/JEANS JACKET": "outerwear", "JEANS/JEANS JACKET": "outwear",
"JEANS/SKIRT": "skirt", "JEANS/SKIRT": "skirt",
"JEANS/TWIN SET": "all-body", "JEANS/TWIN SET": "all-body",
"JEANS/JEANS PANTS": "bottoms", "JEANS/JEANS PANTS": "bottoms",
"JEANS/JEANS SHIRT": "tops", "JEANS/JEANS SHIRT": "tops",
"JEANS/JEANS SKIRT": "skirt", "JEANS/JEANS SKIRT": "skirt",
"JEANS/JEANS VEST": "outerwear", "JEANS/JEANS VEST": "outwear",
"JEANS/JEANS TOP": "tops", "JEANS/JEANS TOP": "tops",
"JEANS/OVERALL": "all-body", "JEANS/OVERALL": "all-body",
"JEANS/PANTS": "bottoms", "JEANS/PANTS": "bottoms",
"JEANS/SHORTS": "bottoms", "JEANS/SHORTS": "bottoms",
"KNIT/CARDGAIN": "outerwear", "KNIT/CARDGAIN": "outwear",
"KNIT/ONE PIECE": "all-body", "KNIT/ONE PIECE": "all-body",
"KNIT/CARDIGAN": "outerwear", "KNIT/CARDIGAN": "outwear",
"KNIT/BRA TOP": "tops", "KNIT/BRA TOP": "tops",
"KNIT/JACKET": "outerwear", "KNIT/JACKET": "outwear",
"KNIT/PULLOVER": "tops", "KNIT/PULLOVER": "tops",
"KNIT/DRESS": "all-body", "KNIT/DRESS": "all-body",
"KNIT/CAPE": "outerwear", "KNIT/CAPE": "outwear",
"KNIT/KNIT TOP": "tops", "KNIT/KNIT TOP": "tops",
"KNIT/CARDGIAN": "outerwear", "KNIT/CARDGIAN": "outwear",
"KNIT/PANTS": "bottoms", "KNIT/PANTS": "bottoms",
"KNIT/GILET": "outerwear", "KNIT/GILET": "outwear",
"KNIT/LONG TOP": "tops", "KNIT/LONG TOP": "tops",
"KNIT/SKIRT": "skirt", "KNIT/SKIRT": "skirt",
"KNIT/SHORTS": "bottoms", "KNIT/SHORTS": "bottoms",
"KNIT/SWEATER": "tops", "KNIT/SWEATER": "tops",
"KNIT/SWEATPR": "tops", "KNIT/SWEATPR": "tops",
"KNIT/VEST": "outerwear", "KNIT/VEST": "outwear",
"KNIT/TANK": "tops", "KNIT/TANK": "tops",
"KNIT/TOP": "tops", "KNIT/TOP": "tops",
"KNIT/TUTA": "all-body", "KNIT/TUTA": "all-body",
"KNIT/TWIN SET": "all-body", "KNIT/TWIN SET": "all-body",
"LEATHER/BLAZER": "outerwear", "LEATHER/BLAZER": "outwear",
"LEATHER/LEATHER BLAZER": "outerwear", "LEATHER/LEATHER BLAZER": "outwear",
"LEATHER/LEATHER COAT": "outerwear", "LEATHER/LEATHER COAT": "outwear",
"LEATHER/LEATHER PANTS": "bottoms", "LEATHER/LEATHER PANTS": "bottoms",
"LEATHER/HAT": "hats", "LEATHER/HAT": "hats",
"LEATHER/LEATHER BLOUSE": "tops", "LEATHER/LEATHER BLOUSE": "tops",
"LEATHER/LEATHER DRESS": "all-body", "LEATHER/LEATHER DRESS": "all-body",
"LEATHER/LEATHER SHORTS": "bottoms", "LEATHER/LEATHER SHORTS": "bottoms",
"LEATHER/LEATHER SKIRT": "skirt", "LEATHER/LEATHER SKIRT": "skirt",
"LEATHER/LEATHER GILET": "outerwear", "LEATHER/LEATHER GILET": "outwear",
"LEATHER/LEATHER JACKET": "outerwear", "LEATHER/LEATHER JACKET": "outwear",
"LEATHER/LEATHER VEST": "outerwear", "LEATHER/LEATHER VEST": "outwear",
"LEATHER/LEATHERSKIRT": "skirt", "LEATHER/LEATHERSKIRT": "skirt",
"LEATHER/LEATHERSHIRT": "tops", "LEATHER/LEATHERSHIRT": "tops",
"LEATHER/LEATHER TOP": "tops", "LEATHER/LEATHER TOP": "tops",
@@ -167,57 +167,57 @@ category_mapping = {
"ONE PIECE/OVERALL": "all-body", "ONE PIECE/OVERALL": "all-body",
"ONE PIECE/SWIMSUIT": "all-body", "ONE PIECE/SWIMSUIT": "all-body",
"ONE PIECE/TWIN SET": "all-body", "ONE PIECE/TWIN SET": "all-body",
"OUTERWEAR/CARDIGAN": "outerwear", "outwear/CARDIGAN": "outwear",
"OUTERWEAR/BLAZER": "outerwear", "outwear/BLAZER": "outwear",
"OUTERWEAR/DOWN COAT": "outerwear", "outwear/DOWN COAT": "outwear",
"OUTERWEAR/": "outerwear", "outwear/": "outwear",
"OUTERWEAR/DOWN JACKET": "outerwear", "outwear/DOWN JACKET": "outwear",
"OUTERWEAR/COAT": "outerwear", "outwear/COAT": "outwear",
"OUTERWEAR/DOWN GILET": "outerwear", "outwear/DOWN GILET": "outwear",
"OUTERWEAR/CAPE": "outerwear", "outwear/CAPE": "outwear",
"OUTERWEAR/DOWN VEST": "outerwear", "outwear/DOWN VEST": "outwear",
"OUTERWEAR/FUR JACKET": "outerwear", "outwear/FUR JACKET": "outwear",
"OUTERWEAR/FUR VEST": "outerwear", "outwear/FUR VEST": "outwear",
"OUTERWEAR/CAPE COAT": "outerwear", "outwear/CAPE COAT": "outwear",
"OUTERWEAR/LEATHER VEST": "outerwear", "outwear/LEATHER VEST": "outwear",
"OUTERWEAR/PULLOVER": "outerwear", "outwear/PULLOVER": "outwear",
"OUTERWEAR/GILET": "outerwear", "outwear/GILET": "outwear",
"OUTERWEAR/FUR COAT": "outerwear", "outwear/FUR COAT": "outwear",
"OUTERWEAR/QUILTED JACKET": "outerwear", "outwear/QUILTED JACKET": "outwear",
"OUTERWEAR/SPORTSWEAR JACKET": "outerwear", "outwear/SPORTSWEAR JACKET": "outwear",
"OUTERWEAR/JACKET": "outerwear", "outwear/JACKET": "outwear",
"OUTERWEAR/FUR GILET": "outerwear", "outwear/FUR GILET": "outwear",
"OUTERWEAR/SHIRT": "tops", "outwear/SHIRT": "tops",
"OUTERWEAR/TWIN SET": "all-body", "outwear/TWIN SET": "all-body",
"OUTERWEAR/LEATHER GILET": "outerwear", "outwear/LEATHER GILET": "outwear",
"OUTERWEAR/JEANS JACKET": "outerwear", "outwear/JEANS JACKET": "outwear",
"OUTERWEAR/TRENCH COAT": "outerwear", "outwear/TRENCH COAT": "outwear",
"OUTERWEAR/LEATHER COAT": "outerwear", "outwear/LEATHER COAT": "outwear",
"OUTERWEAR/ONE PIECE": "all-body", "outwear/ONE PIECE": "all-body",
"OUTERWEAR/LEATHER JACKET": "outerwear", "outwear/LEATHER JACKET": "outwear",
"OUTERWEAR/SCARF": "accessories", "outwear/SCARF": "accessories",
"OUTERWEAR/SOCKS": "accessories", "outwear/SOCKS": "accessories",
"OUTERWEAR/TOP": "tops", "outwear/TOP": "tops",
"OUTERWEAR/UNIFORM": "outerwear", "outwear/UNIFORM": "outwear",
"OUTERWEAR/VEST": "outerwear", "outwear/VEST": "outwear",
"OUTERWEAR/WINDBREAKER": "outerwear", "outwear/WINDBREAKER": "outwear",
"OUTERWEAR/WINDERBREAKER": "outerwear", "outwear/WINDERBREAKER": "outwear",
"OUTPRWEAR/BLAZER": "outerwear", "OUTPRWEAR/BLAZER": "outwear",
"SHOES/PUMPS": "shoes", "SHOES/PUMPS": "shoes",
"SHOES/BOOTS": "shoes", "SHOES/BOOTS": "shoes",
"SHOES/SHOES": "shoes", "SHOES/SHOES": "shoes",
"SHOES/SANDALS": "shoes", "SHOES/SANDALS": "shoes",
"TOP/BLOUSE": "tops", "TOP/BLOUSE": "tops",
"TOP/BLAZER": "outerwear", "TOP/BLAZER": "outwear",
"TOP/BAR TOP": "tops", "TOP/BAR TOP": "tops",
"TOP/BRA TOP": "tops", "TOP/BRA TOP": "tops",
"TOP/CARDIGAN": "outerwear", "TOP/CARDIGAN": "outwear",
"TOP/BODY": "all-body", "TOP/BODY": "all-body",
"TOP/COAT": "outerwear", "TOP/COAT": "outwear",
"TOP/PANTS": "bottoms", "TOP/PANTS": "bottoms",
"TOP/DOWN VEST": "outerwear", "TOP/DOWN VEST": "outwear",
"TOP/JACKET": "outerwear", "TOP/JACKET": "outwear",
"TOP/GILET": "outerwear", "TOP/GILET": "outwear",
"TOP/TWIN SET": "tops", "TOP/TWIN SET": "tops",
"TOP/LONG SLEEVES TEE": "tops", "TOP/LONG SLEEVES TEE": "tops",
"TOP/LONG TOP": "tops", "TOP/LONG TOP": "tops",
@@ -225,7 +225,7 @@ category_mapping = {
"TOP/VEST": "tops", "TOP/VEST": "tops",
"TOP/ONE PIECE": "all-body", "TOP/ONE PIECE": "all-body",
"TOP/TANK": "tops", "TOP/TANK": "tops",
"TOP/PULLOVER": "outerwear", "TOP/PULLOVER": "outwear",
"TOP/TOP": "tops", "TOP/TOP": "tops",
"TOP/OVERALL": "all-body", "TOP/OVERALL": "all-body",
"TOP/UNDERWEAR": "bottoms", "TOP/UNDERWEAR": "bottoms",

2
app/service/attribute_recognition/service.py
View File

@@ -136,7 +136,7 @@ class AttributeRecognition:
present_dict = self.get_attribute(attr_model_path, attr_description, sample) present_dict = self.get_attribute(attr_model_path, attr_description, sample)
attr_dict = Merge(attr_dict, present_dict) attr_dict = Merge(attr_dict, present_dict)
elif category == 'outerwear': elif category == 'outwear':
attr_dict = {'Item': 'outer'} attr_dict = {'Item': 'outer'}
for i in range(len(args.outwear_discription_list)): for i in range(len(args.outwear_discription_list)):