- Libraries
- BiRefNet
How to use ZhengPeng7/BiRefNet_lite-matting with BiRefNet:
# Option 1: use with transformers
from transformers import AutoModelForImageSegmentation
birefnet = AutoModelForImageSegmentation.from_pretrained("ZhengPeng7/BiRefNet_lite-matting", 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-matting") - Transformers
How to use ZhengPeng7/BiRefNet_lite-matting with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("image-segmentation", model="ZhengPeng7/BiRefNet_lite-matting", trust_remote_code=True) # pip install -U transformers accelerate
# Load model directly
from transformers import AutoModelForImageSegmentation
model = AutoModelForImageSegmentation.from_pretrained("ZhengPeng7/BiRefNet_lite-matting", trust_remote_code=True, device_map="auto") - Transformers.js
How to use ZhengPeng7/BiRefNet_lite-matting with Transformers.js:
// npm i @huggingface/transformers
import { pipeline } from '@huggingface/transformers';
// Allocate pipeline
const pipe = await pipeline('image-segmentation', 'ZhengPeng7/BiRefNet_lite-matting'); - Notebooks
- Google Colab
- Kaggle
