Spaces:
Configuration error
Configuration error
Update segmentation.py
Browse files- 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=
|
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))
|