altayavci commited on
Commit
bdf8261
1 Parent(s): d03ca8a

Update segmentation.py

Browse files
Files changed (1) hide show
  1. segmentation.py +1 -1
segmentation.py CHANGED
@@ -62,7 +62,7 @@ def get_blurred_mask(img: Image, body_part_id: int):
62
  crop_mask_binary = crop_mask_np > 128
63
 
64
  dilated_mask = binary_dilation(
65
- crop_mask_binary, iterations=25)
66
  dilated_mask = Image.fromarray((dilated_mask * 255).astype(np.uint8))
67
  dilated_mask_blurred = dilated_mask.filter(
68
  ImageFilter.GaussianBlur(radius=4))
 
62
  crop_mask_binary = crop_mask_np > 128
63
 
64
  dilated_mask = binary_dilation(
65
+ crop_mask_binary, iterations=10)
66
  dilated_mask = Image.fromarray((dilated_mask * 255).astype(np.uint8))
67
  dilated_mask_blurred = dilated_mask.filter(
68
  ImageFilter.GaussianBlur(radius=4))