Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing
    • Website
      • Tasks
      • HuggingChat
      • Collections
      • Languages
      • Organizations
    • Community
      • Blog
      • Posts
      • Daily Papers
      • Learn
      • Discord
      • Forum
      • GitHub
    • Solutions
      • Team & Enterprise
      • Hugging Face PRO
      • Enterprise Support
      • Inference Providers
      • Inference Endpoints
      • Storage Buckets

  • Log In
  • Sign Up

camenduru
/
RMBG-2.0

Image Segmentation
Transformers
PyTorch
ONNX
Safetensors
Transformers.js
remove background
background
background-removal
Pytorch
vision
legal liability
custom_code
Model card Files Files and versions
xet
Community

Instructions to use camenduru/RMBG-2.0 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Transformers

    How to use camenduru/RMBG-2.0 with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("image-segmentation", model="camenduru/RMBG-2.0", trust_remote_code=True)
    # Load model directly
    from transformers import AutoModelForImageSegmentation
    model = AutoModelForImageSegmentation.from_pretrained("camenduru/RMBG-2.0", trust_remote_code=True, dtype="auto")
  • Transformers.js

    How to use camenduru/RMBG-2.0 with Transformers.js:

    // npm i @huggingface/transformers
    import { pipeline } from '@huggingface/transformers';
    
    // Allocate pipeline
    const pipe = await pipeline('image-segmentation', 'camenduru/RMBG-2.0');
  • Notebooks
  • Google Colab
  • Kaggle
RMBG-2.0
5.37 GB
Ctrl+K
Ctrl+K
  • 1 contributor
History: 2 commits
camenduru's picture
camenduru
thanks to briaai ❤
00e2d9a verified 5 months ago
  • onnx
    thanks to briaai ❤ 5 months ago
  • .gitattributes
    1.78 kB
    thanks to briaai ❤ 5 months ago
  • BiRefNet_config.py
    298 Bytes
    thanks to briaai ❤ 5 months ago
  • README.md
    8.49 kB
    thanks to briaai ❤ 5 months ago
  • birefnet.py
    91.4 kB
    thanks to briaai ❤ 5 months ago
  • collage5.png
    4.52 MB
    xet
    thanks to briaai ❤ 5 months ago
  • config.json
    405 Bytes
    thanks to briaai ❤ 5 months ago
  • diagram1.png
    21.3 kB
    thanks to briaai ❤ 5 months ago
  • model.safetensors
    885 MB
    xet
    thanks to briaai ❤ 5 months ago
  • preprocessor_config.json
    391 Bytes
    thanks to briaai ❤ 5 months ago
  • pytorch_model.bin

    Detected Pickle imports (4)

    • "collections.OrderedDict",
    • "torch.LongStorage",
    • "torch._utils._rebuild_tensor_v2",
    • "torch.FloatStorage"

    What is a pickle import?

    885 MB
    xet
    thanks to briaai ❤ 5 months ago
  • t4.png
    2.16 MB
    xet
    thanks to briaai ❤ 5 months ago