ComfyUI-RMBG: 完美人物摳圖,背景移除,精準識別分割
ComfyUI-RMBG是什么?
ComfyUI-RMBG 是一個基于 ai 技術的圖片處理工具,主要用于背景移除和物體分割。它集成了RMBG-2.0、InSPYRENET 和 BEN 等多種先進的模型,可以高效、精準地完成摳圖任務。
ComfyUI-RMBG核心功能:
背景去除(RMBG Node):
能使用多種模型,像RMBG - 2.0、INSPYRENET、BEN、BEN2等。有多種背景選項可選,并且支持批量處理。
RMBG - 2.0模型按照BiRefNet架構構建,在復雜環境里做背景去除時精度高,邊緣細節也能保留。
目標分割(Segment Node):
可以通過文本提示來做目標檢測,支持標簽風格和自然語言輸入,利用SAM(Segment Anything Model)實現高精度分割。
另外,這個項目能手動下載和安裝不少模型,例如RMBG - 2.0、INSPYRENET、BEN、BEN2、BiRefNet、SAM和GroundingDINO等。這些模型在圖像處理場景里各有長處:
RMBG - 2.0:基于BiRefNet架構開發,用于復雜場景的背景提取。有多種背景選項(如透明、黑色、白色),邊緣質量優化后摳圖更自然。
InSPYRENET:主要針對視頻背景去除,批量處理效率高,能調節閾值改善背景效果。
BEN模型:小巧高效,適合算力低的設備,精度較高且計算資源消耗少 。
ComfyUI-RMBG技術要求
需要安裝 ComfyUI 和 Python 3.10+。
依賴的 Python 包包括 torch、torchvision、Pillow、numpy 等。
ComfyUI-RMBG安裝方法
1. 通過 ComfyUI-Manager 安裝
在 ComfyUI-Manager 中搜索 ComfyUI-RMBG 并安裝。安裝完成后,運行以下命令安裝依賴項:
./ComfyUI/python_embeded/python -m pip install -r requirements.txt
2. 手動克隆倉庫
將倉庫克隆到 ComfyUI 的 custom_nodes 文件夾:
cd ComfyUI/custom_nodes git clone https://github.com/1038lab/ComfyUI-RMBG
3. 安裝依賴項。
4. 手動下載模型
首次使用時,模型會自動下載到 ComfyUI/models/RMBG/ 文件夾。
用戶也可以手動下載模型文件,并將其放置在指定目錄,例如:
RMBG-2.0 模型放置在 /ComfyUI/models/RMBG/RMBG-2.0。
INSPYRENET 模型放置在 /ComfyUI/models/RMBG/INSPYRENET。
其他模型的下載鏈接和存放路徑可參考項目文檔。
ComfyUI-RMBG使用場景
電商產品攝影:一鍵摳圖,將產品從復雜背景中提取出來,更換為統一背景。
視頻編輯:支持批量處理視頻,快速更換背景。
AI 換裝和動畫制作:結合其他插件實現背景更換和角色換裝。
藝術創作:在低算力設備上使用輕量級模型(如 BEN),專注于創意表達。
GitHub 倉庫地址:https://github.com/1038lab/ComfyUI-RMBG