Image Segmentation
Transformers
PyTorch
ONNX
Safetensors
Transformers.js
SegformerForSemanticSegmentation
remove background
background
background-removal
Pytorch
vision
legal liability
custom_code
Instructions to use briaai/RMBG-1.4 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use briaai/RMBG-1.4 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True)# Load model directly from transformers import AutoModelForImageSegmentation model = AutoModelForImageSegmentation.from_pretrained("briaai/RMBG-1.4", trust_remote_code=True, dtype="auto") - Transformers.js
How to use briaai/RMBG-1.4 with Transformers.js:
// npm i @huggingface/transformers import { pipeline } from '@huggingface/transformers'; // Allocate pipeline const pipe = await pipeline('image-segmentation', 'briaai/RMBG-1.4'); - Notebooks
- Google Colab
- Kaggle
The result has a black line on the outer contour
#46
by duhaifeng19900711 - opened
You could consider adding some morphological operations on the envelope if the problem is "local".
BTW, If you have enough data, you could try fintuning.
Feel free to join our developer community on Discord to discuss this further.
https://discord.gg/HqnJya2ZHy
Hello me iran
hi @duhaifeng19900711 ,
thanks for your comment,
the problem is mostly visual, depends on the image viewer.
anyway we fixed it and now you shouldn't see a dark silhouette.
origubany changed discussion status to closed




