✂ RMBG-2.0 (BiRefNet) 极速智能抠图工具
基于RMBG-2.0(BiRefNet)目前最强开源抠图模型开发的本地智能抠图工具,支持一键去除图片背景并生成透明背景PNG文件,内置标准图像预处理与原始尺寸还原逻辑,抠图精度高、边缘过渡自然。适配GPU(CUDA)硬件加速,纯本地推理无网络依赖,保障图片隐私安全;搭配Streamlit宽屏双列可视化界面,操作零门槛,支持蒙版查看、抠图结果一键下载,是日常设计、素材处理、图片编辑的高效本地抠图解决方案。
项目简介
本工具基于AI-ModelScope官方RMBG-2.0(BiRefNet)图像分割模型开发,该模型为当前开源领域抠图效果最优的模型之一,可精准分离图像主体与背景,对毛发、半透明物体等边缘细节处理表现优异。工具严格遵循模型训练时的标准预处理流程(1024×1024尺寸缩放、归一化),推理后自动将蒙版还原至图片原始尺寸,避免抠图结果拉伸失真。
工具适配CUDA/CPU双设备,优先使用GPU加速推理,抠图耗时短;通过@st.cache_resource缓存模型,仅首次启动加载一次,后续操作秒响应。基于Streamlit搭建双列极简界面,原始图片与处理结果分区展示,支持JPG/PNG/JPEG格式图片上传,抠图完成后可直接下载透明背景PNG文件,同时提供生成的Alpha蒙版查看功能,兼顾实用性与调试性。纯本地运行无图片上传、无隐私泄露风险,无抠图次数限制,满足日常批量抠图、素材处理需求。
快速启动
启动成功后,控制台将输出访问地址,通过浏览器访问即可进入抠图界面:
操作指南
界面整体布局
工具采用宽屏双列极简布局,所有操作均在浏览器中完成,无需命令行,核心分为两大区域,分区清晰、操作直观,无冗余元素:
- 左列: 图片上传与原始展示区,包含文件上传框、原始图片预览区、一键抠图主按钮,明确标注支持的图片格式(JPG/PNG/JPEG);
- 右列: 处理结果与下载区,包含抠图结果预览区、Alpha蒙版查看扩展栏、透明背景PNG一键下载按钮,实时展示抠图耗时。
顶部显示工具名称及核心特性(最强开源抠图模型、本地推理、隐私安全),模型加载失败时将显示明确错误提示及解决方案,新手友好。
核心操作步骤
上传待抠图图片
点击左列「选择一张图片 (支持 JPG/PNG)」文件上传框,选择本地需要去除背景的图片,支持格式:JPG、JPEG、PNG;图片上传成功后,左列将自动预览原始图片,按容器宽度自适应展示,保留图片原始比例。
一键执行智能抠图
确认图片上传成功后,点击左列** 开始抠图**主按钮(蓝色primary样式,通栏显示),系统将进入抠图处理状态:
- 右列将显示「✂ AI 正在精准分离背景...」加载提示;
- 模型将自动完成图片预处理→GPU/CPU推理→蒙版生成→尺寸还原→透明背景合成全流程;
- 抠图完成后,右列将自动预览抠图结果,并标注实际处理耗时(精确到0.01秒)。
查看结果与下载透明背景图片
- 查看抠图结果:右列预览区展示的是带透明背景的图像效果,可直观查看主体与背景分离效果,边缘细节还原精准;
- 查看Alpha蒙版(可选):点击右列「查看蒙版 (Mask)」扩展栏,可查看模型生成的黑白Alpha蒙版(白色为保留主体,黑色为去除背景),了解模型的分割逻辑;
- 下载结果:点击右列⬇ 下载透明背景 PNG按钮,可将抠图后的透明背景图片保存至本地,文件名为
rmbg_result.png,无水印、无冗余标识,可直接用于设计、素材编辑等场景。