gunnit commited on
Commit
712380b
·
verified ·
1 Parent(s): 54493ff

Update processing/mask.py

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