zl程序教程

您现在的位置是:首页 >  Python

当前栏目

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

2023-03-07 09:02:09 时间

本文教大家通过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)

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

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