patrickligardes commited on
Commit
97b1990
β€’
1 Parent(s): d269fae

Update utils_mask.py

Browse files
Files changed (1) hide show
  1. utils_mask.py +1 -4
utils_mask.py CHANGED
@@ -95,11 +95,8 @@ def get_mask_location(model_type, category, model_parse: Image.Image, keypoint:
95
  parse_mask_legs_dilated = cv2.dilate(parse_mask_legs.astype(np.uint8), np.ones((6, 6), np.uint8), iterations=6)
96
 
97
  # Combine the upper body mask with the dilated leg mask
98
- parse_mask_total = np.maximum(parse_mask_upper, parse_mask_legs_dilated)
99
 
100
- # Apply further processing as needed (e.g., additional masks or refinements)
101
-
102
- parse_mask = np.logical_and(parser_mask_changeable, np.logical_not(parse_mask_total))
103
 
104
 
105
 
 
95
  parse_mask_legs_dilated = cv2.dilate(parse_mask_legs.astype(np.uint8), np.ones((6, 6), np.uint8), iterations=6)
96
 
97
  # Combine the upper body mask with the dilated leg mask
98
+ parse_mask = np.maximum(parse_mask_upper, parse_mask_legs_dilated)
99
 
 
 
 
100
 
101
 
102