Spaces:
Configuration error
Configuration error
Update segmentation.py
Browse files- segmentation.py +1 -1
segmentation.py
CHANGED
@@ -48,7 +48,7 @@ def get_mask(img: Image, body_part_id: int, inverse=False, face=False):
|
|
48 |
return pil_seg
|
49 |
|
50 |
|
51 |
-
def get_cropped(img: Image, body_part_id: int, inverse
|
52 |
|
53 |
pil_seg = get_mask(img, body_part_id, inverse, face)
|
54 |
crop_mask_np = np.array(pil_seg.convert('L'))
|
|
|
48 |
return pil_seg
|
49 |
|
50 |
|
51 |
+
def get_cropped(img: Image, body_part_id: int, inverse:bool, face:bool):
|
52 |
|
53 |
pil_seg = get_mask(img, body_part_id, inverse, face)
|
54 |
crop_mask_np = np.array(pil_seg.convert('L'))
|