Vaibhavnaik12 commited on
Commit
86af068
·
verified ·
1 Parent(s): e2eeaee

Update model/cloth_masker.py

Browse files
Files changed (1) hide show
  1. 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'], # New category for footwear
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'], # New category for footwear
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']