python证件照抠图和换底色

1、去除背景

from removebg import RemoveBg
import os

os.chdir('C:\\Users\\lmy\\Pictures\\Saved Pictures') #更换到照片所在目录

rmbg = RemoveBg("xaQGTCPi6bym8W8qj2EPSUY","error.log")  #引号内是你获取的API,需要自己去注册https://www.remove.bg/
file_path = os.path.join(os.getcwd(),'简历照.jpg') #图片地址
rmbg.remove_background_from_img_file(file_path)
#会自动生成一张去除背景的照片“原照片名.jpg_no_bg.png”

# im1 = Image.open(file_path)  #输入图片
# im1.show()  #展示图片

原图片是这样的(‘简历照.jpg’)
python证件照抠图和换底色_第1张图片
生成的无背景照片是这样的(简历照.jpg_no_bg.png)

python证件照抠图和换底色_第2张图片

2、 填充背景颜色

from PIL import Image #PIL(pillow)库可以完成图像归档和图像处理两方面功能需求
#(1)图像归档:对图像进行批处理、生成图像预览、图像格式转换等;
#(2)图像处理:图像基本处理、像素处理、颜色处理等。

im = Image.open(os.path.join(os.getcwd(),'简历照.jpg_no_bg.png'))  #输入已经去除背景的图片
x,y = im.size
#填充红色背景
p = Image.new('RGBA',im.size,(255,0,0))
p.paste(im,(0,0,x,y),im)
#保存填充后的图片
p.save('red_jianlizhao.png')

换完底色后是这样的(red_jianlizhao.png)
python证件照抠图和换底色_第3张图片

你可能感兴趣的:(python练手小项目,python)