FeyNobg_comfyui

ComfyUI-FeyNobg

基于 BiRefNet 扩展版 FeyNobg 的背景移除 ComfyUI 自定义节点。

简介

FeyNobg 是基于 BiRefNet 扩容并重新训练的版本（24 blocks, 263M 参数），用于自动背景移除、复杂主体识别和边缘保持。

本插件提供两个节点：

• FeyNobg Model Loader — 从 HuggingFace 或 ModelScope 下载并加载模型
• Apply FeyNobg — 执行前景分割和背景移除

安装

将本目录放入 ComfyUI 的 custom_nodes/ 目录，然后安装依赖：

cd ComfyUI/custom_nodes/ComfyUI-FeyNobg
pip install -r requirements.txt

重要： 不要强制重新安装 PyTorch 或 transformers。插件内置兼容的官方 BiRefNet 实现，避免 nobg 对 transformers 5.x 的依赖破坏 ComfyUI 环境。

模型路径

模型自动下载到共享目录 ComfyUI/models/feynobg/，测试和生产环境均可通过 extra_model_paths.yaml 识别。

models/feynobg/
├── huggingface/
│   └── feyninc--FeyNobg/
│       └── model files
└── modelscope/
    └── feyninc--FeyNobg/
        └── model files

使用

基本工作流

FeyNobg Model Loader → Apply FeyNobg → Save Image
Load Image → Apply FeyNobg

Loader 参数

| 参数 | 说明 | |------|------| | source | HuggingFace（feyninc/FeyNobg）或 ModelScope（feyninc/FeyNobg） | | precision | auto/fp32/fp16/bf16 | | device_mode | auto/gpu/cpu | | keep_model_loaded | 是否保留模型在显存 | | force_redownload | 强制重新下载 |

Apply 参数

核心参数：

| 参数 | 说明 | |------|------| | alpha_mode | soft（保留概率）或 binary（二值） | | threshold | 二值阈值 (0-1) | | preserve_original_size | 恢复到原始尺寸 | | invert_mask | 反转 mask | | premultiply_rgb | 预乘 alpha |

增强功能（默认关闭）：

| 参数 | 说明 | |------|------| | horizontal_flip_tta | 水平翻转 TTA | | remove_small_regions | 移除小区域 | | fill_small_holes | 填充小孔洞 | | mask_expand | mask 扩缩 (-64~64 px) | | edge_feather | 边缘羽化 (0~64 px) |

输出

| 输出 | 类型 | 说明 | |------|------|------| | foreground_rgb | IMAGE | 前景 RGB | | alpha_mask | MASK | Alpha 蒙版 [B,H,W] | | rgba_image | IMAGE | RGBA 图像 | | model_info | STRING | 模型信息 |

认证

Token 从环境变量读取，不写入 workflow：

export HF_TOKEN=your_hf_token
export MODELSCOPE_API_TOKEN=your_ms_token

显存

• 模型权重约 1.05 GB
• 建议使用 bf16 或 fp16 精度
• keep_model_loaded=False 时推理后自动卸载

许可证

Apache-2.0