PramaLLC
/

BEN

Image Segmentation

background-remove

mask-generation

Dichotomous image segmentation

background remove

remove background

Model card Files Files and versions

Max Meyer commited on Nov 11, 2024

Commit

0bcab9a

·

verified ·

1 Parent(s): 319ccdc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -20,7 +20,7 @@ file = "./image2.jpg" # input image
 model = model.BEN_Base().to(device).eval() #init pipeline
-model.loadcheckpoints("./BEN/BEN_Base.pth")
 image = Image.open(file)
 mask, foreground = model.inference(image)

 model = model.BEN_Base().to(device).eval() #init pipeline
+model.loadcheckpoints("./BEN/model.pth")
 image = Image.open(file)
 mask, foreground = model.inference(image)