PramaLLC
/

BEN2

Image Segmentation

background-remove

mask-generation

Dichotomous image segmentation

background remove

remove background

model_hub_mixin

pytorch_model_hub_mixin

background removal

background-removal

Model card Files Files and versions

MaxwellMeyer commited on Jan 29, 2025

Commit

5af2efb

·

verified ·

1 Parent(s): e32e557

Update BEN2.py

Files changed (1) hide show

BEN2.py +1 -1

BEN2.py CHANGED Viewed

@@ -1028,7 +1028,7 @@ class BEN_Base(nn.Module):
                 original_image.putalpha(mask)
                 # mask = Image.fromarray(alpha)
-                return foreground
         else:

                 original_image.putalpha(mask)
                 # mask = Image.fromarray(alpha)
+                return original_image
         else: