patrickligardes commited on
Commit
e7828cd
β€’
1 Parent(s): 2d21674

Update utils_mask.py

Browse files
Files changed (1) hide show
  1. 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((8, 8), np.uint8), iterations=8)
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