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
-
Update README.md
Verified signaturee66763e
-
Update requirements.txt
Verified signaturefd83c2e -
Update onnx_run.py
Verified signatured66d54e -
Upload 2 files
Verified signature9305c9a
-
integrate with huggingface (#3)
Verified signature- integrate with huggingface (5ad69f0c63c1820bf9c9486fac0127ded199c980) Co-authored-by: Hafedh Hichri <not-lain@users.noreply.huggingface.co>19d0d22
-
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