Spaces:
Runtime error
Runtime error
Commit ·
ce450be
1
Parent(s): b12ac69
fix type mismatch
Browse files- 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
|