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