Python使用Rembg库去除图片背景

一、引入Rembg库

#库地址
https://github.com/danielgatis/rembg

#CPU使用
pip install rembg # for library
pip install rembg[cli] # for library + cli

#GPU使用（系统支持onnxruntime-gpu）
pip install rembg[gpu] # for library
pip install rembg[gpu,cli] # for library + cli

二、使用

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)

三、效果