altayavci commited on
Commit
52c8d3f
1 Parent(s): 94228f2

Update segmentation.py

Browse files
Files changed (1) hide show
  1. 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=False, face=True):
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'))