1038lab/ComfyUI-RMBG

ComfyUI‑RMBG – 用于 ComfyUI 的背景移除与分割

它是什么 – 一组用于可视化编程界面 ComfyUI 的自定义节点。这些节点封装了大量预训练的视觉模型（RMBG‑2.0、INSPYRENET、BEN/BEN2、BiRefNet、SAM/SAM2/SAM3、Grounding‑DINO、YOLOv8 等），并将其作为可拖拽的模块，可以连接起来以：

• 移除或替换图像背景，
• 分割对象、人脸、服装、时尚物品或任何由文本提示描述的区域，
• 组合掩码、调整大小、拼接图像，以及执行其他实用操作。

为什么重要 – ComfyUI 是一个流行的基于扩散的图像生成工作流引擎。通过添加高质量、GPU 加速的分割节点，创作者可以干净地隔离主体、交换背景，或将掩码输入下游生成模型，而无需离开界面。

主要特性（如 README 中所述）

• 背景移除 – RMBG 节点支持多种模型（RMBG‑2.0、INSPYRENET、BEN、BEN2）。支持批量处理和多种背景填充选项。
• 对象/人脸/时尚分割 – 专门用于通用对象、人脸（19 个面部部位）、服装（18 个类别）和时尚配件的节点。
• 基于提示的分割 – SAM、SAM2 和 SAM3 节点接受标签式或自然语言提示；Grounding‑DINO 为文本查询提供 grounding。
• 实时背景替换 – 将提取的前景与任何用户提供的背景图像交换。
• 实用节点 – 图像/掩码转换器、掩码叠加、掩码增强、图像拼接、裁剪、分辨率选择器、图像比较等。
• VRAM 管理 – unload_model 开关允许用户在节点运行后立即释放 GPU 内存。
• 跨平台支持 – 原生 Apple Silicon (MPS) 加速，兼容 transformers >= 5.0.0。
• 广泛的模型目录 – 包括 SAM‑3 multiplex、BiRefNet 变体、Lucida、SDMatte、Big‑Lama、YOLOv8、Florence‑2 等。

安装 – 三种简单方式：

1. 使用 ComfyUI‑Manager 界面搜索 ComfyUI‑RMBG 并安装。
2. 将仓库克隆到 ComfyUI/custom_nodes 并运行 pip install -r requirements.txt（README 提到 Windows 需要特殊的 PYTHONUTF8=1 标志）。
3. 通过 comfy-cli 命令 comfy node install ComfyUI‑RMBG 安装。

模型处理 – 首次运行会自动将所需模型文件下载到 ComfyUI/models/...。README 还列出了 HuggingFace 上所有支持模型的手动下载链接。

典型工作流

1. 加载图像（LoadImage 节点）。
2. 将其输入到 RMBG 或 Segment 节点。
3. 可选地对掩码进行后处理（MaskEnhancer、MaskOverlay、ImageMaskResize）。
4. 用新图像替换背景（Background Replacement 节点）或将掩码传递给下游扩散节点。

最近更新 – 项目积极维护（v3.2.0 于 2026 年 9 月发布）。亮点包括 SAM3 multiplex 支持、Apple‑Silicon GPU 加速、大幅减少启动时间，以及新的 Resolution Selector 节点。

谁应该使用 – 已经使用 ComfyUI 并需要可靠、高质量分割而无需编写自定义 Python 代码的艺术家、开发者或研究人员。基于节点的方法使过程可视化且可重现。

---

以上所有信息直接取自仓库的 README；未添加任何外部假设。