python证件照抠图和换底色

一、去除背景

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’）

生成的无背景照片是这样的（简历照.jpg_no_bg.png）

二、 填充背景颜色

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）