ZhengPeng7
/

BiRefNet-portrait

Image Segmentation

background-removal

mask-generation

pytorch_model_hub_mixin

model_hub_mixin

Model card Files Files and versions

BiRefNet-portrait

885 MB

1 contributor

History: 22 commits

ZhengPeng7's picture

Fix the no attribute 'all_tied_weights_keys' issue in transformers==5.0.0.

ecdeb62 about 1 month ago

.gitattributes

1.52 kB

initial commit almost 2 years ago
.gitignore

1.91 kB

Add BiRefNet model trained only on portrait segmentation data. over 1 year ago
BiRefNet_config.py

298 Bytes

Add BiRefNet model trained only on portrait segmentation data. over 1 year ago
README.md

4.57 kB

Fix the repo_url in README. about 1 year ago
birefnet.py

92.1 kB

Fix the no attribute 'all_tied_weights_keys' issue in transformers==5.0.0. about 1 month ago
config.json

414 Bytes

Correct the `_name_or_path ` in config.json. over 1 year ago
handler.py

4.71 kB

Align lib_name as birefnet and add inference endpoint option. about 1 year ago
model.safetensors

885 MB
xet

Update portrait seg weights with added mae loss. over 1 year ago
requirements.txt

149 Bytes

Fix the wrong bb setting. Fix the bug by latest transformers and update req.txt. about 1 year ago