Spaces:
Runtime error
Runtime error
Update model/cloth_masker.py
Browse files- 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
|