BEN Base

--- license: apache-2.0 pipeline_tag: image-segmentation tags: - BEN - background-remove - mask-generation - Dichotomous image segmentation - background remove - foreground - background - remove background - pytorch --- ## Model Provided by ParmaLLC The base model is publicly available and free to use for commercial use on HuggingFace: - 🤗 [PramaLLC/BEN](https://huggingface.co/PramaLLC/BEN) ## Quick Start Code (Inside Cloned Repo) ```python import model from PIL import Image import torch device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') file = "./image.png" # input image model = model.BEN_Base().to(device).eval() #init pipeline model.loadcheckpoints("./BEN_Base.pth") image = Image.open(file) mask, foreground = model.inference(image) mask.save("./mask.png") foreground.save("./foreground.png") ``` # BEN SOA Benchmarks on Disk 5k Eval  ### BEN_Base + BEN_Refiner (commercial model please contact us for more information): - MAE: 0.0270 - DICE: 0.8989 - IOU: 0.8506 - BER: 0.0496 - ACC: 0.9740 ### BEN_Base (94 million parameters): - MAE: 0.0309 - DICE: 0.8806 - IOU: 0.8371 - BER: 0.0516 - ACC: 0.9718 ### MVANet (old SOTA): - MAE: 0.0353 - DICE: 0.8676 - IOU: 0.8104 - BER: 0.0639 - ACC: 0.9660 ### BiRefNet(not tested in house): - MAE: 0.038 ### InSPyReNet (not tested in house): - MAE: 0.042 ## Features - Background removal from images - Generates both binary mask and foreground image - CUDA support for GPU acceleration - Simple API for easy integration ## Installation 1. Clone Repo 2. Install requirements.txt

Source context: 27 downloads · 0 likes · Pipeline image-segmentation · Repo juba7/BEN_Base