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/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",

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)
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)):