十行Python代码搞定替换证件照背景颜色

原创

测试加

发布于 2023-02-05 23:32:49

8400

发布于 2023-02-05 23:32:49

文章被收录于专栏：用户4624600的专栏用户4624600的专栏

本文教大家通过Python程序替换证件照背景颜色，以后更换证件照背景就不会再苦恼了。

思路：

1、先去掉原照片的背景颜色

2、再添上新的背景颜色

步骤很简单，思路清晰，操作起来也很简单，十行代码就可以搞定，这么聪明的你肯定没问题。

那咱们开始吧！！！！

先去掉原照片的背景颜色

这里我们要装一个模块 backgroundremover

安装命令

pip3 install backgroundremover

## （报错了不用管）因为要访问谷歌文件没下下来 下面这个就是要下载的

这里装完了还不能用，初次使用需要下载模型（大概170m），下载地址是google，因此需要能够访问google，才能下载。

不过不用担心，号主已经把模型下载好了

在公众号后台回复：u2net

然后把这个文件放到安装用户.u2net目录下

ok问题解决了

下面开始我们的表演:

import os
in_path = "d:/zhu.jpg"
# 去掉背景颜色
os.system('backgroundremover -i "'+str(in_path)+'"  -o "d:\cg_output.jpg"')

in_path是原照片的路径、cg_output.jpg是去掉背景后的照片

添加新背景颜色

# 加上背景颜色
no_bg_image = Image.open("d:\cg_output.jpg")
color = ""
x, y = no_bg_image.size
new_image = Image.new('RGBA', no_bg_image.size, color=color)
new_image.paste(no_bg_image, (0, 0, x, y), no_bg_image)
new_image.save(out_path)

out_path 是替换背景颜色后的照片路径，color 是要替换的新颜色，填上对应的英文即可，比如 天空蓝：deepskyblue

颜色参数表

完整代码

import os
from PIL import Image
# 输入
in_path = "d：\cui.jpg"
# 输出
out_path = "d:\out.png"
# 要替换的背景颜色
color = "deepskyblue"
# 红：red、蓝：blue、黑：black、白：white

# 去掉背景颜色
os.system('backgroundremover -i "'+str(in_path)+'"  -o "d:\cg_output.jpg"')
# 加上背景颜色
no_bg_image = Image.open("d:\cg_output.jpg")
x, y = no_bg_image.size
new_image = Image.new('RGBA', no_bg_image.size, color=color)
new_image.paste(no_bg_image, (0, 0, x, y), no_bg_image)
new_image.save(out_path)

好了，也介绍完了，大家也可以自己练练

附上我的效果图哈哈）—- 见笑了