ComfyUI-RMBG 开源项目使用与启动教程

ComfyUI-RMBG 开源项目使用与启动教程

ComfyUI-RMBG A ComfyUI custom node designed for advanced image background removal and object, face, clothes, and fashion segmentation, utilizing multiple models including RMBG-2.0, INSPYRENET, BEN, BEN2, SAM, and GroundingDINO. 项目地址: https://gitcode.com/gh_mirrors/co/ComfyUI-RMBG

1. 项目介绍

ComfyUI-RMBG 是一个基于 ComfyUI 的自定义节点，它专门用于高级图像背景移除和对象、面部、衣物以及时尚元素的分割。该项目集成了多种模型，包括 RMBG-2.0、INSPYRENET、BEN、BEN2、BiRefNet 模型、SAM 和 GroundingDINO，为用户提供了强大的图像处理功能。

2. 项目快速启动

安装

方法1：通过 ComfyUI-Manager 安装

1. 在 ComfyUI-Manager 中搜索 Comfyui-RMBG 并安装。
2. 在 ComfyUI-RMBG 文件夹中安装 requirements.txt：

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

方法2：通过克隆仓库安装

1. 切换到 ComfyUI 的 custom_nodes 目录下：

   cd ComfyUI/custom_nodes
   

2. 克隆仓库：

   git clone https://github.com/1038lab/ComfyUI-RMBG
   

3. 在 ComfyUI-RMBG 文件夹中安装 requirements.txt：

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

方法3：通过 Comfy CLI 安装

1. 确保 comfy-cli 已安装。
2. 如果未安装 ComfyUI，先安装 ComfyUI：

   comfy install
   

3. 使用以下命令安装 ComfyUI-RMBG：

   comfy node install ComfyUI-RMBG
   

4. 在 ComfyUI-RMBG 文件夹中安装 requirements.txt：

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

模型下载

首次使用自定义节点时，模型将自动下载到 ComfyUI/models/RMBG/ 目录下。以下是手动下载模型的链接：

• RMBG-2.0 模型：下载链接
• INSPYRENET 模型：下载链接
• BEN 模型：下载链接
• BEN2 模型：下载链接
• BiRefNet-HR 模型：下载链接

将下载的文件放到对应的文件夹中。

示例代码

以下是使用 ComfyUI-RMBG 的一个简单示例：

# 示例代码，实际使用时需要根据实际情况调整
from ComfyUI_RMBG import RMBGNode

# 创建 RMBG 节点实例
node = RMBGNode()

# 加载图像
image = node.load_image("path/to/your/image.jpg")

# 背景移除
background_removed_image = node.remove_background(image)

# 保存图像
node.save_image(background_removed_image, "output.jpg")

3. 应用案例和最佳实践

背景移除

适用于需要移除图片背景的各种场景，如广告设计、图像合成等。

面部分割

在面部识别、美容修饰等领域有广泛应用。

衣物分割

为电商、服装设计等行业提供精确的衣物分割服务。

时尚元素分割

帮助分析时尚元素，为时尚行业提供数据支持。

4. 典型生态项目

• 图像处理工具集：整合了多种图像处理功能的工具集，用于各种图像编辑需求。
• 电商平台：利用衣物和时尚元素分割功能，为商品展示提供更专业的图像处理。
• 虚拟试衣应用：结合面部和衣物分割技术，提供虚拟试衣体验。

ComfyUI-RMBG A ComfyUI custom node designed for advanced image background removal and object, face, clothes, and fashion segmentation, utilizing multiple models including RMBG-2.0, INSPYRENET, BEN, BEN2, SAM, and GroundingDINO. 项目地址: https://gitcode.com/gh_mirrors/co/ComfyUI-RMBG