ComfyUI-RMBG: идеальное вырезание объектов, удаление фона и точная сегментация 192

小编 ·

ComfyUI-RMBG: идеальное вырезание объектов, удаление фона и точная сегментация

ComfyUI-RMBG — инструмент для обработки изображений на основе ИИ, предназначенный для удаления фона и сегментации объектов. Он объединяет несколько передовых моделей, таких как RMBG-2.0, InSPYRENET и BEN, обеспечивая эффективное и точное вырезание.

ComfyUI-RMBG: идеальное вырезание объектов, удаление фона и точная сегментация

Что такое ComfyUI-RMBG?

ComfyUI-RMBG — это инструмент для обработки изображений на основе технологии искусственного интеллекта,主要用于背景移除和物体分割。它集成了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)实现高精度分割。

Основные функции ComfyUI-RMBG.webp

另外,这个项目能手动下载和安装不少模型,例如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.webp

Способ установки ComfyUI-RMBG

1. Установка через ComfyUI-Manager

В ComfyUI-Manager найдите ComfyUI-RMBG и установите его.安装完成后,运行以下命令安装依赖项:

./ComfyUI/python_embeded/python -m pip install -r requirements.txt

2. Ручное клонирование репозитория

Клонируйте репозиторий в папку custom_nodes ComfyUI:

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),专注于创意表达。

Сценарии использования ComfyUI-RMBG.webp

Адрес репозитория на GitHub:https://github.com/1038lab/ComfyUI-RMBG