FeyNoBg – Feyn Labs 开源的自动背景去除模型

FeyNoBg是什么

FeyNoBg 是 Feyn Labs 推出的开源自动背景去除模型，基于 BiRefNet 架构深度改进，拥有 2.63 亿参数。模型在 UHRSD-TE、HRSOD-TE 等 8 项基准测试中，有 4 项取得 SOTA 成绩，其余 4 项与最优结果差距不足 2%。团队同时开源了 NoBg Python 库，提供统一的模型运行与训练接口，支持 Hugging Face 生态，采用 Apache-2.0 协议。

FeyNoBg的主要功能

• 高精度背景去除：自动识别图像前景并生成 Alpha 遮罩，支持复杂边缘如头发、毛发、细线等。
• 超高分辨率支持：覆盖 4K、8K 场景的 UHRSD-TE 基准测试表现领先。
• 图像与视频抠图：支持单图、批量图片及视频帧序列的背景去除。
• 开源可训练：通过 NoBg 库可在自定义数据集上微调，支持替换损失函数与调整分辨率。
• Hugging Face 集成：模型与处理器均兼容 transformers 生态，支持一键下载与推送。

FeyNoBg的技术原理

• BiRefNet 双模块架构：定位模块负责前景识别，重建模块负责边界追踪，两者互补协作。
• 第三阶段深度扩展：将特征提取器第三阶段从 18 个块扩展至 24 个块，参数从 2.22 亿增至 2.63 亿，保留原有预训练权重，仅新增块从零训练，避免知识遗忘。
• 多样化数据混合：整合 10 个数据集的 26,100 张图像，涵盖拥挤场景、伪装、高分辨率、肖像、动漫等类型，每来源上限 4,000 张防止单一数据主导。
• 统一标注格式：将分割数据集的前景掩码与抠图数据集的 Alpha 通道统一转换为二值前景掩码，使训练目标一致。
• S-measure 评估：通过 0-1 区间的 S-measure 指标同时奖励主体完整性与形状保真度。

微信关注回复“开源”，加入AI开源项目交流群

如何使用FeyNoBg

• 安装环境：通过 pip install nobg 安装库，并确保 Python ≥ 3.10 与 torch ≥ 2.0 环境就绪。
• 加载模型：用 AutoModel.from_pretrained("feyninc/FeyNobg") 从 Hugging Face 下载并加载预训练权重。
• 加载处理器：使用 AutoProcessor.from_pretrained("feyninc/FeyNobg") 初始化图像预处理模块。
• 读取图像：将目标图片转换为 RGB 格式，确保输入符合模型要求。
• 预处理：调用 processor 对图像进行尺寸调整与归一化，生成模型所需的张量输入。
• 模型推理：在 torch.inference_mode() 下将 pixel_values 传入模型，获取背景去除的原始输出。
• 后处理遮罩：用 post_process_alpha_matting 将模型输出还原为原始图像尺寸的 Alpha 遮罩。
• 生成结果：调用 cutout 方法将原图前景与 Alpha 遮罩合成，保存为透明 PNG 文件。

FeyNoBg的核心优势

• SOTA 性能：在 8 项基准中 4 项领先，超高分辨率场景优势显著。
• 知识保留训练：扩展网络深度时保留原始权重，避免灾难性遗忘。
• 工程优化：NoBg 库在批大小 1/2/4 下均优于原始 BiRefNet 实现，吞吐量更高、延迟更低、显存占用更少。
• 生态兼容：深度集成 Hugging Face transformers，支持 Trainer、Hub 推送与半精度推理。

FeyNoBg的项目地址

• 项目官网：https://usefeyn.com/blog/feynobg/
• GitHub仓库：https://github.com/feyninc/nobg
• HuggingFace模型库：https://huggingface.co/feyninc/FeyNobg

FeyNoBg的应用场景

• 电商产品图处理：FeyNoBg 可批量去除商品背景并生成透明底图，直接用于电商平台上架展示。
• 广告营销设计：设计师可使用 FeyNoBg 快速提取人物或产品主体，将其合成到新背景中以制作宣传海报。
• 影视后期抠像：FeyNoBg 支持对 4K 与 8K 视频帧进行高精度背景去除，能够替代传统的绿幕拍摄与后期抠像流程。
• 证件照精修：用户可借助 FeyNoBg 自动分离人物与复杂背景，实现换底色操作并对发丝等边缘细节进行优化。
• AI 训练数据生成：FeyNoBg 可为图像合成、生成模型等下游任务批量产出高质量的前景掩码标注数据。