视觉/图像重磅干货,第一时间送达!
背景介绍
开源库rembg,让你轻松实现去除图片背景的效果。github地址:
https://github.com/danielgatis/rembg
准备工作
Python版本要求>3.7,<3.13,直接pip安装即可:
pip install rembg -i https://pypi.tuna.tsinghua.edu.cn/simple
测试代码:
from PIL import Image
from rembg import remove
input = Image.open("1.jpg")
output = remove(input)
output.save("car.png")
第一次运行会先下载预训练的模型,下载地址:
https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net.onnx
可以手动下载模型,放到:C:\Users\你的用户名\.u2ne文件夹下
应用一:制作证件照
使用上面的代码,加载一张正脸照片,下面是本文使用的测试图,图片来源于网络,如有侵权请联系删除。
from PIL import Image
from rembg import remove
input = Image.open("2.jpg")
output = remove(input)
output.save("BackgroundRemovedFlower.png")
设置对应的图片名,运行上面的代码,可以得到下面的结果:
得到一张透明背景的图片,复制到PPT中,双击图片--开始--形状填充--选择背景颜色,蓝底红底任你选择,最后右键保存即可!
应用二:更改目标背景
还是上面的代码,替换测试图片:
分割结果如下:
得到一张透明背景的图片,复制到PPT中,再加一张背景图置于底部,移动刚刚生成的图像即可,最后截图保存。
当然,上面的2个案例都是借助PPT完成,大家也可以尝试使用OpenCV来完成。