From 474001da6b9da5d6293ed58f6d5f08ba44c4766a Mon Sep 17 00:00:00 2001 From: Akasei Date: Mon, 16 Mar 2026 22:14:21 +0800 Subject: [PATCH] feat(rembg): switch background removal to bria-rmbg model Replace default u2net with bria-rmbg-2.0 for better quality. BackgroundRemover now accepts model_name param (defaults to 'bria-rmbg'). Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> --- hy3dshape/hy3dshape/rembg.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hy3dshape/hy3dshape/rembg.py b/hy3dshape/hy3dshape/rembg.py index 6247f06..bd94de3 100644 --- a/hy3dshape/hy3dshape/rembg.py +++ b/hy3dshape/hy3dshape/rembg.py @@ -17,8 +17,8 @@ from rembg import remove, new_session class BackgroundRemover(): - def __init__(self): - self.session = new_session() + def __init__(self, model_name: str = "bria-rmbg"): + self.session = new_session(model_name) def __call__(self, image: Image.Image): output = remove(image, session=self.session, bgcolor=[255, 255, 255, 0])