nobg How to use nobg/BiRefNet with nobg:
pip install nobg
# Option 1: use via the predict method
from nobg import AutoModel, AutoProcessor
model = AutoModel.from_pretrained("nobg/BiRefNet").eval()
processor = AutoProcessor.from_pretrained("nobg/BiRefNet")
cutout = model.predict(processor, "image.jpg") # Option 2: use the model and processor directly
import torch
from loadimg import load_img
from nobg import AutoModel, AutoProcessor
model = AutoModel.from_pretrained("nobg/BiRefNet").eval()
processor = AutoProcessor.from_pretrained("nobg/BiRefNet")
image = load_img("image.jpg").convert("RGB")
inputs = processor(image, return_tensors="pt")
with torch.no_grad():
outputs = model(pixel_values=inputs["pixel_values"])
alpha = processor.post_process_alpha_matting(outputs, target_sizes=[(image.height, image.width)])[0]
processor.cutout(image, alpha).save("output.png")
