Spaces:
Running
on
Zero
Running
on
Zero
patrickligardes
commited on
Commit
β’
e7828cd
1
Parent(s):
2d21674
Update utils_mask.py
Browse files- utils_mask.py +1 -1
utils_mask.py
CHANGED
@@ -90,7 +90,7 @@ def get_mask_location(model_type, category, model_parse: Image.Image, keypoint:
|
|
90 |
(parse_array == 13).astype(np.float32)
|
91 |
|
92 |
# Fill gaps between legs
|
93 |
-
leg_gap_filled = cv2.dilate(leg_gap_mask.astype(np.uint8), np.ones((
|
94 |
|
95 |
# Add the gap-filled mask to the existing parse_mask
|
96 |
parse_mask += leg_gap_filled
|
|
|
90 |
(parse_array == 13).astype(np.float32)
|
91 |
|
92 |
# Fill gaps between legs
|
93 |
+
leg_gap_filled = cv2.dilate(leg_gap_mask.astype(np.uint8), np.ones((10, 10), np.uint8), iterations=10)
|
94 |
|
95 |
# Add the gap-filled mask to the existing parse_mask
|
96 |
parse_mask += leg_gap_filled
|