Mask Generation
BEN2
Safetensors
BEN2
Dichotomous image segmentation
background remove
background-remove
model_hub_mixin
pytorch_model_hub_mixin
Instructions to use aldigobbler/ben2_30m_test with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- BEN2
How to use aldigobbler/ben2_30m_test 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("aldigobbler/ben2_30m_test") model.to("cuda").eval() foreground = model.inference(image) - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -11,7 +11,9 @@ tags:
|
|
| 11 |
---
|
| 12 |
# BEN2 pruning test # 1
|
| 13 |
Original parameters: 94,134,658
|
|
|
|
| 14 |
Pruned parameters: 36,275,254
|
|
|
|
| 15 |
Parameter reduction: 61.5%
|
| 16 |
|
| 17 |
- Swin Transformer depths: [2, 2, 18, 2] → [2, 2, 6, 2]
|
|
|
|
| 11 |
---
|
| 12 |
# BEN2 pruning test # 1
|
| 13 |
Original parameters: 94,134,658
|
| 14 |
+
|
| 15 |
Pruned parameters: 36,275,254
|
| 16 |
+
|
| 17 |
Parameter reduction: 61.5%
|
| 18 |
|
| 19 |
- Swin Transformer depths: [2, 2, 18, 2] → [2, 2, 6, 2]
|