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

briaai
/
RMBG-1.4

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

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
RMBG-1.4
180 MB
Ctrl+K
Ctrl+K
  • 13 contributors
History: 87 commits
linoyts's picture
linoyts HF Staff
Update README.md
847d7a6 verified over 2 years ago
  • .gitattributes
    1.85 kB
    Upload t4.png over 2 years ago
  • README.md
    4.87 kB
    Update README.md over 2 years ago
  • briarmbg.py
    12.9 kB
    Upload briarmbg.py over 2 years ago
  • example_inference.py
    1.19 kB
    Update example_inference.py over 2 years ago
  • example_input.jpg
    327 kB
    Rename example_image.jpg to example_input.jpg over 2 years ago
  • model.pth

    Detected Pickle imports (4)

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

    What is a pickle import?

    177 MB
    xet
    Upload model.pth over 2 years ago
  • requirements.txt
    66 Bytes
    Update requirements.txt over 2 years ago
  • results.png
    1.25 MB
    xet
    Upload results.png over 2 years ago
  • t4.png
    2.16 MB
    xet
    Upload t4.png over 2 years ago
  • utilities.py
    980 Bytes
    Upload utilities.py over 2 years ago