Image Segmentation
BEN2
ONNX
Safetensors
PyTorch
BEN2
background-remove
mask-generation
Dichotomous image segmentation
background remove
foreground
background
remove background
model_hub_mixin
pytorch_model_hub_mixin
background removal
background-removal
Instructions to use PramaLLC/BEN2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- BEN2
How to use PramaLLC/BEN2 with BEN2:
import requests from PIL import Image from ben2 import AutoModel url = "https://huggingface.co/datasets/mishig/sample_images/resolve/main/teapot.jpg" image = Image.open(requests.get(url, stream=True).raw) model = AutoModel.from_pretrained("PramaLLC/BEN2") model.to("cuda").eval() foreground = model.inference(image) - Notebooks
- Google Colab
- Kaggle
Commit History
-
integrate with huggingface
Verified signatureThis pr will add the necessary metadata to your repository you can test these changes **before** merging using the following code ``` pip install -e "git+https://github.com/PramaLLC/BEN2.git#egg=ben2" ``` ```python from ben2 import BEN_Base model = BEN_Base.from_pretrained("PramaLLC/BEN2", revision="refs/pr/3") # revision is used to load from a pull request or a previous commit ``` I also updated your readme file to include the latest changes.5ad69f0
-
Update requirements.txt
Verified signature97f07dc -
Update BEN2.py
Verified signatureb210846
-
Update README.md
Verified signature11be3ff -
Update README.md
Verified signature3b5223c -
Update inference.py
Verified signatureb2cbb2b -
Update BEN2.py
Verified signature0004261 -
Update README.md
Verified signatured8358da -
Update inference.py
Verified signature764ae38 -
Update README.md
Verified signatureb9aa096 -
Update BEN2.py
Verified signaturefe0c9f4 -
Update BEN2.py
Verified signature5af2efb -
Update BEN2.py
Verified signaturee32e557
-
Upload 12 files
Verified signaturecefc11a
-
initial commit
Verified signaturec83f50d