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

Commit

b2cbb2b

·

verified ·

1 Parent(s): 0004261

Update inference.py

Files changed (1) hide show

inference.py +1 -2

inference.py CHANGED Viewed

@@ -11,7 +11,6 @@ model = BEN2.BEN_Base().to(device).eval() #init pipeline
 model.loadcheckpoints("./BEN2_Base.pth")
 image = Image.open(file)
-mask, foreground = model.inference(image)
-mask.save("./mask.png")
 foreground.save("./foreground.png")

 model.loadcheckpoints("./BEN2_Base.pth")
 image = Image.open(file)
+foreground = model.inference(image)
 foreground.save("./foreground.png")