Rembg

Rembg是一个开源的Python软件包，用于自动去除图片背景。 默认使用U²-Net进行前景检测，使用PyMatting进行抠图。

简介

安装

仅使用cpu计算，安装rembg版本：

pip install rembg

使用GPU加速，

Python中使用

使用pillow库进行，PIL的图片的操作。

from rembg import remove
from PIL import Image

input_path = 'input.png'
output_path = 'output.png'

input = Image.open(input_path)
output = remove(input)
output.save(output_path)

参数

名称	描述	示例
alpha_matting	是否使用alpha matting参数，默认False，
alpha_matting_foreground_threshold	前景阈值，默认240
alpha_matting_background_threshold	背景阈值，默认10
alpha_matting_erode_size	默认10
only_mask	默认False

资源

官网

• Rembg 官网：https://github.com/danielgatis/rembg
• Rembg 源代码：https://github.com/danielgatis/rembg

网站

• https://github.com/xuebinqin/U-2-Net
• https://github.com/pymatting/pymatting