mohammadhakimi commited on
Commit
ce450be
·
1 Parent(s): b12ac69

fix type mismatch

Browse files
Files changed (1) hide show
  1. processing/mask.py +1 -1
processing/mask.py CHANGED
@@ -9,7 +9,7 @@ from DPT.dpt.transforms import PrepareForNet, NormalizeImage, Resize
9
 
10
 
11
  def create_mask(image, blur=False, padding=0):
12
- rm_bg = remove(image, post_process_mask=True, only_mask=True)
13
  rm_bg = Image.fromarray((rm_bg * 255).astype(np.uint8))
14
  rm_bg = rm_bg.resize(image.size, resample=Image.BILINEAR)
15
  # Create a padding of 5 pixels around the object in the mask
 
9
 
10
 
11
  def create_mask(image, blur=False, padding=0):
12
+ rm_bg = remove(np.array(image), post_process_mask=True, only_mask=True)
13
  rm_bg = Image.fromarray((rm_bg * 255).astype(np.uint8))
14
  rm_bg = rm_bg.resize(image.size, resample=Image.BILINEAR)
15
  # Create a padding of 5 pixels around the object in the mask