Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing

  • Log In
  • Sign Up

ZhengPeng7
/
BiRefNet

Image Segmentation
BiRefNet
Safetensors
Transformers
background-removal
mask-generation
Dichotomous Image Segmentation
Camouflaged Object Detection
Salient Object Detection
pytorch_model_hub_mixin
model_hub_mixin
custom_code
Model card Files Files and versions
xet
Community
19

Instructions to use ZhengPeng7/BiRefNet with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • BiRefNet

    How to use ZhengPeng7/BiRefNet with BiRefNet:

    # Option 1: use with transformers
    
    from transformers import AutoModelForImageSegmentation
    birefnet = AutoModelForImageSegmentation.from_pretrained("ZhengPeng7/BiRefNet", trust_remote_code=True)
    
    # Option 2: use with BiRefNet
    
    # Install from https://github.com/ZhengPeng7/BiRefNet
    
    from models.birefnet import BiRefNet
    model = BiRefNet.from_pretrained("ZhengPeng7/BiRefNet")
  • Transformers

    How to use ZhengPeng7/BiRefNet with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("image-segmentation", model="ZhengPeng7/BiRefNet", trust_remote_code=True)
    # Load model directly
    from transformers import AutoModelForImageSegmentation
    model = AutoModelForImageSegmentation.from_pretrained("ZhengPeng7/BiRefNet", trust_remote_code=True, dtype="auto")
  • Notebooks
  • Google Colab
  • Kaggle
BiRefNet
885 MB
Ctrl+K
Ctrl+K
  • 6 contributors
History: 33 commits
osanseviero's picture
osanseviero
Add some metadata
601d8b2 verified almost 2 years ago
  • models
    Move all BiRefNet github codes to the first level directory. almost 2 years ago
  • .gitattributes
    1.52 kB
    initial commit almost 2 years ago
  • .gitignore
    1.91 kB
    For users to load in one key. almost 2 years ago
  • BiRefNet_config.py
    224 Bytes
    For users to load in one key. almost 2 years ago
  • BiRefNet_pipe.py
    242 Bytes
    Move all BiRefNet github codes to the first level directory. almost 2 years ago
  • README.md
    432 Bytes
    Add some metadata almost 2 years ago
  • __init__.py
    213 Bytes
    Move all BiRefNet github codes to the first level directory. almost 2 years ago
  • birefnet.py
    14.7 kB
    Move all BiRefNet github codes to the first level directory. almost 2 years ago
  • config.json
    449 Bytes
    Move all BiRefNet github codes to the first level directory. almost 2 years ago
  • config.py
    8.08 kB
    Move all BiRefNet github codes to the first level directory. almost 2 years ago
  • dataset.py
    6.75 kB
    Move all BiRefNet github codes to the first level directory. almost 2 years ago
  • model.safetensors
    885 MB
    xet
    Push model using huggingface_hub. almost 2 years ago
  • preproc.py
    2.93 kB
    Move all BiRefNet github codes to the first level directory. almost 2 years ago
  • train.sh
    1.42 kB
    Move all BiRefNet github codes to the first level directory. almost 2 years ago
  • utils.py
    2.75 kB
    Move all BiRefNet github codes to the first level directory. almost 2 years ago