Spaces:
Runtime error
Runtime error
Update processing/mask.py
Browse files- 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=
|
| 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)
|