ComfyUI-BEN

ComfyUI-BEN
★ 45

背景移除图像抠图BEN模型ComfyUI扩展
在ComfyUI中集成BEN模型，自动从图像中抠出前景并移除背景，便于合成与后期处理。
💡 在ComfyUI流程中自动去除图像背景以便合成或抠图
🍴 3 Forks💻 Python🔄 2024-12-15
📦
网盘下载
复制链接后前往夸克网盘下载
https://pan.quark.cn/s/9d76119b2ef2
📦 requirements.txt
huggingface_hub
Image
numpy
torch
einops
Pillow
timm
torchvision
ben
📄 README

ComfyUI BEN – Background Erase Network

Remove backgrounds from images with BEN in ComfyUI

Installation

git clone https://github.com/DoctorDiffusion/ComfyUI-BEN.git
cd ComfyUI-BEN
pip install -r requirements.txt

Go to the BEN huggingface page and download BEN_Base.pth, and model.py and place them inside:

...ComfyUI/custom-nodes/ComfyUI-BEN/

The folder should look like this:

└── ComfyUI/custom-nodes/ComfyUI-BEN/
    ├── __init__.py
    ├── background_erase_network.py
    ├── ben.png
    ├── BEN_Base.pth
    ├── model.py
    ├── README.md
    └── requirements.txt

Nodes

Background Erase Network

Outputs image of subject with alpha layer, combine with Image to Mask Node to get alpha mask as well.

Credits

  • comfyanonymous/ComfyUI

  • PramaLLC/BEN

    • ⭐ If you like the project, please give it a star! ⭐

    License

    Apache 2.0 License

    OUTPUTS CLEAR FOR COMMERCIAL USE

    PramaLLC – “You can use our BEN model commercially without any problem. Its under the Apache 2.0 license. The only commercial piece is the BEN+Refiner but the BEN_BASE is perfectly fine for commercial use. :)”