Spaces:
Runtime error
Runtime error
Update model/cloth_masker.py
Browse files- model/cloth_masker.py +3 -3
model/cloth_masker.py
CHANGED
@@ -83,9 +83,9 @@ MASK_CLOTH_PARTS = {
|
|
83 |
'lower': ['Pants', 'Skirt', 'Dress', 'Jumpsuits'],
|
84 |
'overall': ['Upper-clothes', 'Dress', 'Pants', 'Skirt', 'Coat', 'Jumpsuits'],
|
85 |
'inner': ['Upper-clothes'],
|
86 |
-
'outer': ['Coat',]
|
87 |
'bags': ['Bag'], # New category for bags
|
88 |
-
'footwear': ['Left-shoe', 'Right-shoe']
|
89 |
}
|
90 |
MASK_DENSE_PARTS = {
|
91 |
'upper': ['torso', 'big arms', 'forearms'],
|
@@ -94,7 +94,7 @@ MASK_DENSE_PARTS = {
|
|
94 |
'inner': ['torso'],
|
95 |
'outer': ['torso', 'big arms', 'forearms']
|
96 |
'bags': [], # No dense parts for bags
|
97 |
-
'footwear': ['left foot', 'right foot']
|
98 |
}
|
99 |
|
100 |
schp_public_protect_parts = ['Hat', 'Hair', 'Sunglasses', 'Left-shoe', 'Right-shoe', 'Bag', 'Glove', 'Scarf']
|
|
|
83 |
'lower': ['Pants', 'Skirt', 'Dress', 'Jumpsuits'],
|
84 |
'overall': ['Upper-clothes', 'Dress', 'Pants', 'Skirt', 'Coat', 'Jumpsuits'],
|
85 |
'inner': ['Upper-clothes'],
|
86 |
+
'outer': ['Coat',],
|
87 |
'bags': ['Bag'], # New category for bags
|
88 |
+
'footwear': ['Left-shoe', 'Right-shoe'] # New category for footwear
|
89 |
}
|
90 |
MASK_DENSE_PARTS = {
|
91 |
'upper': ['torso', 'big arms', 'forearms'],
|
|
|
94 |
'inner': ['torso'],
|
95 |
'outer': ['torso', 'big arms', 'forearms']
|
96 |
'bags': [], # No dense parts for bags
|
97 |
+
'footwear': ['left foot', 'right foot'] # New category for footwear
|
98 |
}
|
99 |
|
100 |
schp_public_protect_parts = ['Hat', 'Hair', 'Sunglasses', 'Left-shoe', 'Right-shoe', 'Bag', 'Glove', 'Scarf']
|