diff --git a/README.md b/README.md index 757e7f1..f59da52 100644 --- a/README.md +++ b/README.md @@ -50,6 +50,20 @@ ___ Thanks to BiRefNet repo owner [ZhengPeng7/BiRefNet](https://github.com/zhengpeng7/birefnet) +``` +library_name: birefnet +tags: + - background-removal + - mask-generation + - Dichotomous Image Segmentation:二分图像分割(DIS),指分割成前景与背景,二个集合,分割出高精度效果。 + - Camouflaged Object Detection:伪装物体检测(COD),偏工程向,旨在识别“无缝”嵌入到周围环境中的物体,例如野生动物保护、军事侦察或者工业自动化。 + - Salient Object Detection:显著性目标检测(SOD),自动检测图像中最具视觉吸引力的部分。 + - pytorch_model_hub_mixin + - model_hub_mixin +repo_url: https://github.com/ZhengPeng7/BiRefNet +pipeline_tag: image-segmentation +license: mit +``` 部分代码参考了 [ZHO-ZHO-ZHO/ComfyUI-BiRefNet-ZHO](https://github.com/ZHO-ZHO-ZHO/ComfyUI-BiRefNet-ZHO) 感谢! Some of the code references [ZHO-ZHO-ZHO/ComfyUI-BiRefNet-ZHO](https://github.com/ZHO-ZHO-ZHO/ComfyUI-BiRefNet-ZHO) Thanks! diff --git a/nodes.py b/nodes.py index e49ad23..e2a8b37 100644 --- a/nodes.py +++ b/nodes.py @@ -102,6 +102,8 @@ class BiRefNet_Hugo: rembg_list.insert(0, "Auto_DownLoad-ZhengPeng7/BiRefNet") rembg_list.insert(1, "Auto_DownLoad-ZhengPeng7/BiRefNet-DIS5K-TR_TEs") rembg_list.insert(2, "Auto_DownLoad-ZhengPeng7/BiRefNet-COD") + rembg_list.insert(3, "Auto_DownLoad-ZhengPeng7/BiRefNet-HRSOD") + rembg_list.insert(4, "Auto_DownLoad-ZhengPeng7/BiRefNet-portrait") return { "required": { "model": (rembg_list, ),