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
importlib error caused by module name with dot
#44
by rongjhan - opened
AutoModelForImageSegmentation.from_pretrained("briaai/RMBG-1.4",trust_remote_code=True) , error happend when execute from_pretrained
full module path is "transformers_modules.briaai.RMBG-1.4.22532afbdabdc36b2d30a334076720ac72a06f83.MyConfig"
this error is shown in transformers 4.38.2 , i'm not sure whether it occur in older or newer edition , but i don't see anyone point it out
i think the possible way to fix this error is clone whole repository and rename package to make it work
like this discussion do
or is possible remove edition number from repository name ?
origubany changed discussion status to closed