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

Update model/cloth_masker.py

Browse files
Files changed (1) hide show
  1. model/cloth_masker.py +2 -1
model/cloth_masker.py CHANGED
@@ -226,7 +226,8 @@ class AutoMasker:
226
  part_mask_of(['Hair'], schp_atr_mask, ATR_MAPPING)
227
  cloth_protect_area = part_mask_of(PROTECT_CLOTH_PARTS[part]['LIP'], schp_lip_mask, LIP_MAPPING) | \
228
  part_mask_of(PROTECT_CLOTH_PARTS[part]['ATR'], schp_atr_mask, ATR_MAPPING)
229
- accessory_protect_area = part_mask_of((accessory_parts := ['Hat', 'Glove', 'Sunglasses
 
230
 
231
 
232
  # Mask Area
 
226
  part_mask_of(['Hair'], schp_atr_mask, ATR_MAPPING)
227
  cloth_protect_area = part_mask_of(PROTECT_CLOTH_PARTS[part]['LIP'], schp_lip_mask, LIP_MAPPING) | \
228
  part_mask_of(PROTECT_CLOTH_PARTS[part]['ATR'], schp_atr_mask, ATR_MAPPING)
229
+ accessory_protect_area = part_mask_of((accessory_parts := ['Hat', 'Glove', 'Sunglasses'])) | \
230
+ part_mask_of(accessory_parts, schp_atr_mask, ATR_MAPPING)
231
 
232
 
233
  # Mask Area