Duplicated from briaai/RMBG-1.4
How to use wide-video/rmbg-v1.0.0 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-segmentation", model="wide-video/rmbg-v1.0.0", trust_remote_code=True)
# Load model directly from transformers import AutoModelForImageSegmentation model = AutoModelForImageSegmentation.from_pretrained("wide-video/rmbg-v1.0.0", trust_remote_code=True, device_map="auto")
What is a pickle import?
