rembg

项目概述

rembg 是一个移除图片背景的工具，通过内置的 U2Net 等 AI 模型把前景主体从背景中分离出来。它既能当命令行工具用，也能作为 Python 库、HTTP 服务器或 Docker 容器运行，模型在本地推理，图片不必上传第三方服务。

本地 AI 抠图，一条命令把主体从背景里拿出来。

主要功能

• 一键去背景：输入任意图片，自动识别主体并输出透明背景结果
• 多种使用形态：命令行、Python 库、HTTP 服务、Docker 容器均可运行
• 多模型可选：内置 u2net、isnet 等模型，可按人像、物体等场景切换
• CPU 与 GPU：默认 CPU 推理，NVIDIA 显卡可切换 onnxruntime-gpu 加速
• Alpha Matting：开启边缘优化参数，让发丝等细节过渡更自然
• 批量处理：命令行可对整个目录的图片批量去背景

适用人群

• 电商与素材制作人员
• 批量处理图片的内容团队
• 需要集成抠图能力的应用开发者
• 要求本地处理、注重隐私的设计师

快速上手

1. 按硬件选择后端安装

pip install 'rembg[cpu,cli]'

2. 用命令行处理单张图片

rembg i input.png output.png

3. 或在代码中调用

from rembg import remove
from PIL import Image

inp = Image.open('input.png')
remove(inp).save('output.png')

核心优势

• 本地推理：模型在本机运行，图片不出内网
• 形态灵活：从个人脚本到服务化部署，同一套能力多端复用
• 开箱即用：首次运行自动下载模型，无需手工配置环境

系统要求

注意事项

• 首次运行需下载模型：模型文件较大，网络不佳时可提前手动放置到缓存目录
• 效果因图而异：主体与背景颜色相近或边缘复杂的图片，建议尝试不同模型或开启 alpha matting