FIT-Check commited on
Commit
d0afe6a
·
verified ·
1 Parent(s): 68bc001

Update model/cloth_masker.py

Browse files
Files changed (1) hide show
  1. model/cloth_masker.py +2 -2
model/cloth_masker.py CHANGED
@@ -207,7 +207,7 @@ class AutoMasker:
207
  schp_atr_mask = np.array(schp_atr_mask)
208
 
209
  # Strong Protect Area (Hands, Face, Accessory, Feet)
210
- if False and part == "overall":
211
  hands_protect_area = part_mask_of(['hands',], densepose_mask, DENSE_INDEX_MAP)
212
  else:
213
  hands_protect_area = part_mask_of(['hands', 'feet'], densepose_mask, DENSE_INDEX_MAP)
@@ -225,7 +225,7 @@ class AutoMasker:
225
  part_mask_of(['Hair'], schp_atr_mask, ATR_MAPPING)
226
  cloth_protect_area = part_mask_of(PROTECT_CLOTH_PARTS[part]['LIP'], schp_lip_mask, LIP_MAPPING) | \
227
  part_mask_of(PROTECT_CLOTH_PARTS[part]['ATR'], schp_atr_mask, ATR_MAPPING)
228
- if False and part == "overall":
229
  accessory_protect_area = part_mask_of((accessory_parts := ['Hat', 'Glove', 'Sunglasses', 'Bag', 'Scarf', 'Socks']), schp_lip_mask, LIP_MAPPING) | \
230
  part_mask_of(accessory_parts, schp_atr_mask, ATR_MAPPING)
231
  else:
 
207
  schp_atr_mask = np.array(schp_atr_mask)
208
 
209
  # Strong Protect Area (Hands, Face, Accessory, Feet)
210
+ if part == "overall":
211
  hands_protect_area = part_mask_of(['hands',], densepose_mask, DENSE_INDEX_MAP)
212
  else:
213
  hands_protect_area = part_mask_of(['hands', 'feet'], densepose_mask, DENSE_INDEX_MAP)
 
225
  part_mask_of(['Hair'], schp_atr_mask, ATR_MAPPING)
226
  cloth_protect_area = part_mask_of(PROTECT_CLOTH_PARTS[part]['LIP'], schp_lip_mask, LIP_MAPPING) | \
227
  part_mask_of(PROTECT_CLOTH_PARTS[part]['ATR'], schp_atr_mask, ATR_MAPPING)
228
+ if part == "overall":
229
  accessory_protect_area = part_mask_of((accessory_parts := ['Hat', 'Glove', 'Sunglasses', 'Bag', 'Scarf', 'Socks']), schp_lip_mask, LIP_MAPPING) | \
230
  part_mask_of(accessory_parts, schp_atr_mask, ATR_MAPPING)
231
  else: