patrickligardes commited on
Commit
7772536
ยท
verified ยท
1 Parent(s): adfb1f7

Update utils_mask.py

Browse files
Files changed (1) hide show
  1. utils_mask.py +2 -3
utils_mask.py CHANGED
@@ -92,13 +92,12 @@ def get_mask_location(model_type, category, model_parse: Image.Image, keypoint:
92
 
93
  # Fill gaps between legs
94
  leg_mask_filled = cv2.dilate(parse_mask_legs.astype(np.uint8), np.ones((6, 6), np.uint8), iterations=6)
95
- leg_mask_filled = cv2.erode(leg_mask_filled, np.ones((6, 6), np.uint8), iterations=6) # Optional: Erode back to original size
96
 
97
  # Combine filled leg mask with upper body mask
98
  parse_mask = np.maximum(parse_mask_upper, leg_mask_filled)
99
 
100
- # Final dilation to ensure there are no small gaps
101
- parse_mask = cv2.dilate(parse_mask.astype(np.uint8), np.ones((5, 5), np.uint8), iterations=5)
102
 
103
 
104
 
 
92
 
93
  # Fill gaps between legs
94
  leg_mask_filled = cv2.dilate(parse_mask_legs.astype(np.uint8), np.ones((6, 6), np.uint8), iterations=6)
95
+
96
 
97
  # Combine filled leg mask with upper body mask
98
  parse_mask = np.maximum(parse_mask_upper, leg_mask_filled)
99
 
100
+
 
101
 
102
 
103