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

2cd441b

·

verified ·

1 Parent(s): 0bcab9a

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -2,7 +2,7 @@
 license: apache-2.0
 ---
-# BEN - Background Erase Network
 BEN is a deep learning model designed to automatically remove backgrounds from images, producing both a mask and a foreground image.
 - MADE IN AMERICA
@@ -20,7 +20,7 @@ file = "./image2.jpg" # input image
 model = model.BEN_Base().to(device).eval() #init pipeline
-model.loadcheckpoints("./BEN/model.pth")
 image = Image.open(file)
 mask, foreground = model.inference(image)

 license: apache-2.0
 ---
+# BEN - Background Erase Network (Base Model)
 BEN is a deep learning model designed to automatically remove backgrounds from images, producing both a mask and a foreground image.
 - MADE IN AMERICA
 model = model.BEN_Base().to(device).eval() #init pipeline
+model.loadcheckpoints("./BEN/BEN_Base.pth")
 image = Image.open(file)
 mask, foreground = model.inference(image)