FeyNoBg:Feyn Labs开源的新一代高精度AI抠图模型

AI新闻 AI工具箱
61

1. 什么是 FeyNoBg

FeyNoBg 是 Feyn Labs 发布的SOTA级开源自动图像背景移除(AI抠图)模型,配套自研标准化Python开发库 NoBg,整套方案完全开源免费,面向开发者、设计师、AI研究人员提供高精度抠图能力。

该模型基于BiRefNet架构深度优化,专门解决传统抠图模型「主体识别与精细边缘无法兼顾」的痛点,在超高清图像、发丝、伪装物体、拥挤场景等高难度素材中实现行业顶尖分割效果,同时提供统一、高性能的训练与推理工具链,大幅降低图像抠图二次开发门槛。

FeyNoBg(图1)

2. 核心功能特色

2.1 高精度发丝级抠像

同时兼顾全局前景识别细边缘Alpha抠像,可精准还原飘动头发、细金属丝、衣物绒毛、半透明薄纱等复杂边界,输出原图尺寸透明PNG素材。

2.2 超高清图像原生适配

原生支持4K/8K超高分辨率图片处理,在UHRSD-TE、HRSOD-TE高清基准数据集性能大幅领先同类模型,不会出现大图细节丢失、边缘糊化问题。

2.3 开源完整工具链 NoBg

配套自研Python库NoBg,统一封装推理、训练、数据预处理逻辑;相比原生BiRefNet,吞吐量更高、单图延迟更低、GPU显存占用更少,无需适配多套零散代码库。

2.4 轻量化模型扩容、无知识遗忘

仅小幅提升参数量(222M→263M),复用全部预训练权重,仅新增模块从零训练,新增学习容量同时保留原有识别能力,不会出现训练后基础场景精度下滑。

2.5 全场景泛化训练数据集

混合10大类共26100张多领域图像训练,覆盖人像、商品、动漫、伪装物体、拥挤街景、合成素材,低对比度、复杂遮挡场景鲁棒性更强。

2.6 开箱即用+自定义微调双模式

普通用户可直接调用预训练模型一键抠图;开发者可基于NoBg接入自有图像+掩码数据集,快速微调专属行业抠图模型,原生兼容Hugging Face Trainer训练框架。

3. 底层技术细节

3.1 模型架构优化逻辑

基础底座采用BiRefNet双分支双边参考网络:

  1. 定位模块:识别图像全局前景主体,区分前景与大面积背景;

  2. 重建模块:精细化还原物体轮廓、发丝等细微边缘;

  3. 核心升级:特征提取器第三阶段(兼顾全局与局部细节的核心层)模块数量由18块扩充至24块,新增6个未初始化学习模块;

  4. 参数变化:总参数量由222M提升至263M,原有权重全部保留,仅新增模块参与重新训练,避免灾难性遗忘。

3.2 多元化均衡训练策略

  1. 单一数据集缺陷验证:仅使用MaskFactory训练会出现场景偏科,部分基准指标大幅下降;

  2. 混合数据集配置:整合S3OD、MaskFactory、DIS5K、UHRSD等10类数据源,总量26100张;

  3. 数据平衡规则:每个数据集上限4000张,避免大样本数据集主导训练;统一分割掩码、透明度Alpha图为二值掩码,标准化训练监督目标;

  4. 数据筛选:剔除泛化增益弱的ThinObject-5K、HIM-2K、COIFT数据集,保留泛化能力最优组合。

3.3 性能评测指标(S-measure,0~1,数值越高分割越精准)

基于8大权威图像分割基准测试,4项榜单取得最优成绩,剩余4项与行业顶尖模型差距控制在2%以内:

基准数据集 FeyNoBg分数 行业最优竞品 性能差距
UHRSD-TE(超高清图) 0.981 BiRefNet 0.957 FeyNoBg +2.5%
HRSOD-TE(高分辨率目标) 0.983 S3ODNet 0.961 FeyNoBg +2.3%
DIS5K(精细二值分割) 0.961 FlowDIS 0.951 FeyNoBg +1.1%
DAVIS-S(视频静态帧) 0.977 BiRefNet 0.975 FeyNoBg +0.2%
DUTS-TE(通用显著目标) 0.941 S3ODNet 0.949 落后0.8%
COD10K-TE(伪装物体) 0.912 S3ODNet 0.923 落后1.2%
DUT-OMRON(复杂背景) 0.883 S3ODNet 0.898 落后1.7%
CAMO-TE(隐蔽目标) 0.887 BiRefNet 0.904 落后1.9%

3.4 NoBg库底层优化

  1. 推理流水线内置图像缩放、归一化、蒙版还原逻辑,自动匹配原图尺寸输出透明Alpha通道;

  2. 批量推理优化:batch size 1/2/4全场景下,吞吐量提升、单图耗时降低、峰值显存占用少于原生BiRefNet实现;

  3. 训练层封装:内置损失函数、数据整理器,一行代码接入Hugging Face训练器。

FeyNoBg(图2)

4. 落地应用场景

  1. 电商商品素材处理:服装、3C、日用品批量抠白底图,处理反光、透明包装、复杂配饰边缘;

  2. 人像内容创作:写真、短视频人像、证件照发丝精细抠图,适配直播、自媒体封面;

  3. 影视/平面设计:婚纱、毛发、动态模糊人物抠像,替代PS手动钢笔、通道抠图;

  4. 工业图像分割:高精度产品检测、微小零部件提取、伪装目标识别;

  5. 动漫/二次元素材:手绘插画、动漫人物批量去背景,适配二次元内容平台;

  6. 私有化企业部署:企业本地批量抠图系统、小程序/网站抠图API、离线素材处理工具;

  7. AI科研实验:图像分割、显著性检测、抠图算法对比、自定义模型微调研究。

5. 完整使用方法

5.1 环境安装

执行命令一键安装官方库:

pip install nobg

5.2 预训练模型推理(一键抠图输出透明PNG)

import torch
from loadimg import load_img
from nobg import AutoModel, AutoProcessor

# 加载预训练模型与处理器
model = AutoModel.from_pretrained("feyninc/FeyNobg").eval()
processor = AutoProcessor.from_pretrained("feyninc/FeyNobg")

# 读取本地图片
image = load_img("input.jpg").convert("RGB")
inputs = processor(image, return_tensors="pt")

# 推理生成透明蒙版
with torch.inference_mode():
  outputs = model(pixel_values=inputs["pixel_values"])
  alpha = processor.post_process_alpha_matting(
    outputs,
    target_sizes=[(image.height, image.width)],
  )[0]
# 保存透明背景抠图结果
processor.cutout(image, alpha).save("output.png")

5.3 自有数据集自定义微调训练

原生兼容Hugging Face Trainer,仅需准备「原图+黑白掩码」数据集:

from nobg import AutoModel, AutoProcessor
from transformers import Trainer, TrainingArguments

model = AutoModel.from_pretrained("feyninc/FeyNobg")
processor = AutoProcessor.from_pretrained("feyninc/FeyNobg")

# 数据整理函数
def collate(examples):
  batch = processor(
    images=[ex["image"] for ex in examples],
    segmentation_maps=[ex["mask"].convert("L") for ex in examples],
    return_tensors="pt",
  )
  return {"pixel_values": batch["pixel_values"], "labels": batch["labels"]}

# 训练配置
trainer = Trainer(
  model=model,
  args=TrainingArguments(output_dir="outputs", learning_rate=2e-5),
  train_dataset=dataset,
  data_collator=collate,
)
trainer.train()

5.4 在线快速体验

无需本地部署,直接访问Hugging Face在线空间上传图片实时抠图。

6. 竞品对比

选取行业主流开源抠图模型 Rembg、原生BiRefNet 与 FeyNoBg 对比:

对比维度 FeyNoBg BiRefNet(原生) Rembg
基础架构 BiRefNet深度扩容优化 原版双边参考网络 集成多轻量化分割模型
参数量 263M 222M 约176M(RMBG核心)
高清4K/8K适配 原生优化,4项高清基准第一 支持但细节损耗更大 高分辨率易糊边
发丝/精细物体精度 最优,混合多源数据训练 优秀,单一数据集训练偏科 简单场景可用,发丝易残缺
配套开发库 NoBg统一封装,推理训练一体 无官方统一库,代码零散 仅封装推理,不支持自定义训练
GPU显存占用 更低,批量推理优化 基准占用更高 轻量但精度妥协
自定义微调支持 原生支持,兼容Hugging Face Trainer 需自行搭建训练流水线 不支持微调自有数据集
开源完整度 模型、训练库、文档全开源 仅开源模型权重 仅推理工具开源
适用场景 专业设计、工业、科研、高分辨率素材 通用高精度分割,泛化弱 轻量批量简单图片抠图

7. 常见问题问答(FAQ)

Q1:FeyNoBg 和原生 BiRefNet 核心区别是什么?

A:FeyNoBg 在BiRefNet基础上扩容核心特征层、复用全部预训练权重,搭配10类混合均衡数据集训练,新增配套NoBg标准化开发库;原生BiRefNet仅单一架构,无统一训练推理工具,单一数据集训练易出现场景偏科,高分辨率素材性能弱于FeyNoBg。

Q2:FeyNoBg 是否可以商用,有无版权限制?

A:模型与NoBg库完全开源,可免费商用;引用时需标注官方提供的文献引用格式,同时遵循底层BiRefNet原作者开源协议。

Q3:本地运行最低硬件要求是什么?

A:CPU可完成推理,速度较慢;推荐搭载8GB及以上显存NVIDIA GPU实现快速批量处理;微调训练建议12GB以上显存显卡。

Q4:NoBg库相比原生代码有什么性能提升?

A:同批量尺寸下吞吐量更高、单图推理延迟更低、峰值GPU显存占用更少,内置标准化预处理、后处理逻辑,无需手动编写蒙版还原代码。

Q5:模型可以处理视频素材抠图吗?

A:原生仅支持单张静态图像抠图,论文评测基准DAVIS-S为视频单帧测试;连续视频抠图需自行增加帧时序追踪逻辑扩展。

Q6:训练自定义数据集时,掩码文件格式要求是什么?

A:统一使用黑白二值掩码图片,前景白色、背景黑色;库内部会自动将不同类型标注转换为统一训练目标,无需手动调整。

Q7:为什么混合多数据集训练,还要限制每个数据集最大4000张图片?

A:若不限制样本数量,数据量更大的数据集会主导训练过程,造成模型只擅长某一类素材,丢失泛化能力;限制样本上限可平衡各类场景训练权重。

Q8:免费在线演示空间有无图片尺寸、次数限制?

A:Hugging Face在线空间存在免费额度、单图分辨率限制;无限制批量处理、高清导出建议本地部署NoBg库运行预训练模型。

8. 官方链接

  1. 官方博客文档:https://usefeyn.com/blog/feynobg/

  2. Hugging Face:https://huggingface.co/feyninc/FeyNobg

  3. GitHub仓库地址:https://github.com/feyninc/nobg

9. 总结

FeyNoBg 是基于成熟BiRefNet架构迭代优化的新一代开源高精度背景移除模型,通过适度扩容特征提取网络、均衡多元混合数据集解决了传统抠图模型主体识别与精细边缘无法兼顾的核心痛点,在超高清、发丝、伪装物体等高难度图像分割场景取得四项SOTA基准最优成绩;配套自研NoBg标准化Python库统一封装推理与训练流程,在运行效率、显存占用、二次开发便捷度上全面优于同类开源方案,完整开源无商用限制,覆盖设计师、企业开发者、AI科研人员的离线批量抠图、私有化部署、自定义模型微调全场景需求,是兼顾精度、性能与开发友好度的一站式AI抠图解决方案。

打赏
THE END
作者头像
AI工具箱
一个喜欢收集AI工具的小萌新