KawasakiAkasei/Hunyuan3D_2.1_Low_VRAM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(rembg): switch background removal to bria-rmbg model

Browse Source 
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>
This commit is contained in:
Akasei
2026-03-16 22:14:21 +08:00
parent 76c36e53eb
commit 474001da6b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
hy3dshape/hy3dshape/rembg.py
View File

@@ -17,8 +17,8 @@ from rembg import remove, new_session
class BackgroundRemover(): class BackgroundRemover():
def __init__(self): def __init__(self, model_name: str = "bria-rmbg"):
self.session = new_session() self.session = new_session(model_name)
def __call__(self, image: Image.Image): def __call__(self, image: Image.Image):
output = remove(image, session=self.session, bgcolor=[255, 255, 255, 0]) output = remove(image, session=self.session, bgcolor=[255, 255, 255, 0])