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

  • Log In
  • Sign Up

ZhengPeng7
/
BiRefNet_lite

Image Segmentation
BiRefNet
Safetensors
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

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

  • Libraries
  • BiRefNet

    How to use ZhengPeng7/BiRefNet_lite with BiRefNet:

    # Option 1: use with transformers
    
    from transformers import AutoModelForImageSegmentation
    birefnet = AutoModelForImageSegmentation.from_pretrained("ZhengPeng7/BiRefNet_lite", 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_lite")
  • Notebooks
  • Google Colab
  • Kaggle
BiRefNet_lite
178 MB
Ctrl+K
Ctrl+K
  • 1 contributor
History: 17 commits
ZhengPeng7's picture
ZhengPeng7
Fix the no attribute 'all_tied_weights_keys' issue in transformers==5.0.0.
7838f1c 3 months ago
  • .gitattributes
    1.52 kB
    initial commit almost 2 years ago
  • BiRefNet_config.py
    298 Bytes
    Initial uploading of the BiRefNet for general use with swin_v1_tiny as the backbone. almost 2 years ago
  • README.md
    8.72 kB
    Add half precision setting in README. over 1 year ago
  • birefnet.py
    92.1 kB
    Fix the no attribute 'all_tied_weights_keys' issue in transformers==5.0.0. 3 months ago
  • config.json
    410 Bytes
    Update config.json over 1 year ago
  • handler.py
    4.71 kB
    Align lib_name as birefnet and add inference endpoint option. over 1 year ago
  • model.safetensors
    178 MB
    xet
    Initial uploading of the BiRefNet for general use with swin_v1_tiny as the backbone. almost 2 years ago
  • requirements.txt
    149 Bytes
    Fix the bug by latest transformers and update req.txt. about 1 year ago