
rembg人像分割3行代码实现发丝级背景去除【免费下载链接】rembgRembg is a tool to remove images background项目地址: https://gitcode.com/GitHub_Trending/re/rembg上周客户丢来一批证件照要换底色初稿交回去被退回头发边缘全是锯齿鞋子还被我抠没了。这是人像分割里很典型的坑——通用背景去除工具处理物品没问题一碰到人发丝和边缘过渡就开始掉链子。rembg 里正好有一个人像专用模型可以解决这个问题下面按我们实际跑通的路径走一遍。为什么选 birefnet-portrait 做人像分割与通用模型和手动PS的差距通用分割模型比如 u2net处理人像实测最明显的问题是底部边缘小腿以下整段被判成背景连鞋都直接消失。手动 PS 一张图也许只要几分钟但批量处理时手会酸质量也靠手感。birefnet-portrait 是专为人像分割做的 session它的权重文件名带 epoch_150说明这份检查点是针对人像数据训练到第 150 轮才取的。模型源码与权重在哪里所有模型统一注册在 rembg/sessions/init.pynew_session按名字找到对应类。birefnet-portrait 的下载逻辑在 birefnet_portrait.py权重通过网络拉取并做 md5 校验checksum_disabled开关可跳过校验。安装与最小可运行示例安装并跑通第一次背景去除pip install rembg[cpu] pillowfrom PIL import Image from rembg import new_session, remove # 创建人像专用会话首次运行会自动下载权重 session new_session(birefnet-portrait) with Image.open(examples/girl-1.jpg) as img: result remove(img, sessionsession) # 去除背景返回透明PNG result.save(output_portrait.png)首次运行会下载 onnx 权重并缓存在用户主目录之后直接复用不用反复下载。效果对比与调参同一张图的两个模型效果上面两张是同一张输入分别用 u2net 和 birefnet-portrait 处理的结果差异集中在两处u2net 版本整只鞋的区域丢失腿部在膝盖处被截断人像模型版本则保留了完整身形头发外轮廓上人像模型边缘更连续头侧的细碎发丝没有被切成一条线。当然左侧袖口附近还能看到一点半透明残留这就是下一步调参要解决的。证件照底色与发丝边缘调参证件照自动换背景第一步就是定底色。客户要求红底蓝底时在 remove 一步直接合成bgcolor接收 RGBA 四元组result remove(img, sessionsession, bgcolor(255, 0, 0, 255)) # 红底轮廓上有明显锯齿或边缘残留时打开掩码后处理形态学开运算加高斯平滑实现见 bg.pyresult remove(img, sessionsession, post_process_maskTrue)发丝、眼镜反光这类难区再叠加 alpha matting估算每个像素透明度的算法。它速度偏慢建议只对最终交付的图开result remove( img, sessionsession, alpha_mattingTrue, alpha_matting_foreground_threshold270, )模型选型速查模型适合场景关键差异坑/注意birefnet-portrait证件照、人像特写最准速度较慢首次需联网下载权重u2net-human-seg全身人像、动作照片快全身轮廓够用发丝细节弱于 portraitu2net默认任意主体背景去除最快最通用人物底部边缘易被截断u2netp实时处理、网页展示轻量小模型比 u2net 明显更快精度最低sam需要精确交互的区域支持点选指定要传坐标用法更复杂完整模型清单见 USAGE.md。落地时容易踩的坑批量处理时复用会话如果要处理整个文件夹别在循环里反复new_session——session 会把权重常驻内存反复创建既慢又吃内存而且remove默认每次调用都新建会话这正是多图处理的性能瓶颈。正确做法是创建一次 session、传给所有调用USAGE.md 里有多图批量的写法可直接抄。CPU线程数与GPU加速只装pip install rembg不带后端时import 会直接报 No onnxruntime backend found必须显式装rembg[cpu]或rembg[gpu]NVIDIA 卡。new_session会读取OMP_NUM_THREADS环境变量来设置 onnxruntime 的线程数批量处理想控制 CPU 并发设置这个变量即可。边缘仍不理想时的兜底alpha matting 之后发丝还有白边或锯齿可以先调高alpha_matting_foreground_threshold收紧不确定区域实在不行就only_maskTrue只输出掩码人工修完再自己合成——别指望模型一步到位。场景是人像特写、证件照、成品化的自动换底色birefnet-portrait 值得花时间如果是主体杂七杂八的通用场景批量处理u2net 加后处理就够了人像模型的精度收益有限。人像分割模型源码birefnet_portrait.py 核心入口rembg/init.py 使用文档USAGE.md 效果样例tests/results/ 测试脚本tests/test_remove.py【免费下载链接】rembgRembg is a tool to remove images background项目地址: https://gitcode.com/GitHub_Trending/re/rembg创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考