Spaces:
Running
on
Zero
Running
on
Zero
Update utils_mask.py
Browse files- 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 |
-
|
96 |
|
97 |
# Combine filled leg mask with upper body mask
|
98 |
parse_mask = np.maximum(parse_mask_upper, leg_mask_filled)
|
99 |
|
100 |
-
|
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 |
|