Spaces:
Running
on
Zero
Running
on
Zero
shape
Browse files- demo/rm_bg.py +1 -0
demo/rm_bg.py
CHANGED
|
@@ -8,6 +8,7 @@ def rm_bg(img, use_sam=False):
|
|
| 8 |
img = img.resize((512, 512))
|
| 9 |
output = rembg.remove(img)
|
| 10 |
mask = np.array(output)[:, :, 3]
|
|
|
|
| 11 |
|
| 12 |
# use sam for mask refinement
|
| 13 |
if use_sam:
|
|
|
|
| 8 |
img = img.resize((512, 512))
|
| 9 |
output = rembg.remove(img)
|
| 10 |
mask = np.array(output)[:, :, 3]
|
| 11 |
+
print(mask.shape)
|
| 12 |
|
| 13 |
# use sam for mask refinement
|
| 14 |
if use_sam:
|