ZhengPeng7
/

BiRefNet-portrait

Image Segmentation

background-removal

mask-generation

pytorch_model_hub_mixin

model_hub_mixin

Model card Files Files and versions Community

BiRefNet-portrait

Ctrl+K

Ctrl+K

1 contributor

History: 18 commits

ZhengPeng7's picture

Fix the wrong bb setting. Fix the bug by latest transformers and update req.txt.

e77ea69 3 months ago

.gitattributes

1.52 kB

initial commit about 1 year ago
.gitignore

1.91 kB

Add BiRefNet model trained only on portrait segmentation data. 10 months ago
BiRefNet_config.py

298 Bytes

Add BiRefNet model trained only on portrait segmentation data. 10 months ago
README.md

4.57 kB

Fix the repo_url in README. 3 months ago
birefnet.py

92.1 kB

Fix the wrong bb setting. Fix the bug by latest transformers and update req.txt. 3 months ago
config.json

414 Bytes

Correct the `_name_or_path ` in config.json. 9 months ago
handler.py

4.71 kB

Align lib_name as birefnet and add inference endpoint option. 4 months ago
model.safetensors

885 MB
LFS

Update portrait seg weights with added mae loss. 10 months ago
requirements.txt

149 Bytes

Fix the wrong bb setting. Fix the bug by latest transformers and update req.txt. 3 months ago