minjung-s
commited on
Commit
•
00eaae9
1
Parent(s):
c5c6bad
0311
Browse files- model/cloth_masker2.py +1 -0
model/cloth_masker2.py
CHANGED
|
@@ -304,6 +304,7 @@ class AutoMasker:
|
|
| 304 |
body_protect_area = part_mask_of([], schp_lip_mask, LIP_MAPPING) | part_mask_of([], schp_atr_mask, ATR_MAPPING)
|
| 305 |
hair_protect_area = part_mask_of(['Hair'], schp_lip_mask, LIP_MAPPING) | \
|
| 306 |
part_mask_of(['Hair'], schp_atr_mask, ATR_MAPPING)
|
|
|
|
| 307 |
cloth_protect_area = part_mask_of(PROTECT_CLOTH_PARTS['overall']['LIP'], schp_lip_mask, LIP_MAPPING) | \
|
| 308 |
part_mask_of(PROTECT_CLOTH_PARTS['overall']['ATR'], schp_atr_mask, ATR_MAPPING)
|
| 309 |
accessory_protect_area = part_mask_of((accessory_parts := ['Hat', 'Glove', 'Sunglasses', 'Bag', 'Left-shoe', 'Right-shoe', 'Scarf', 'Socks']), schp_lip_mask, LIP_MAPPING) | \
|
|
|
|
| 304 |
body_protect_area = part_mask_of([], schp_lip_mask, LIP_MAPPING) | part_mask_of([], schp_atr_mask, ATR_MAPPING)
|
| 305 |
hair_protect_area = part_mask_of(['Hair'], schp_lip_mask, LIP_MAPPING) | \
|
| 306 |
part_mask_of(['Hair'], schp_atr_mask, ATR_MAPPING)
|
| 307 |
+
|
| 308 |
cloth_protect_area = part_mask_of(PROTECT_CLOTH_PARTS['overall']['LIP'], schp_lip_mask, LIP_MAPPING) | \
|
| 309 |
part_mask_of(PROTECT_CLOTH_PARTS['overall']['ATR'], schp_atr_mask, ATR_MAPPING)
|
| 310 |
accessory_protect_area = part_mask_of((accessory_parts := ['Hat', 'Glove', 'Sunglasses', 'Bag', 'Left-shoe', 'Right-shoe', 'Scarf', 'Socks']), schp_lip_mask, LIP_MAPPING) | \
|