利用Python的removebg实现抠图加背景颜色!~~
转载原作者声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
https://blog.csdn.net/Alicia_N/article/details/97164090
在原作者的代码基础上,进行了修改添加,能跑出来不过效果有些欠缺(新手小白,各位大神请忽略。。。):
1.先导入需要的python库,我的是py.37的,自带了pillow(相当于PIL),在anaconda prompt里面输入pip install removebg就成功安装了removebg库了。。
import os
from removebg import RemoveBg
from PIL import Image
2.接下来我们需要获取API(在removebg的官网注册申请就可以了,我前几天申的时候是只绑定了邮箱,链接:https://www.remove.bg/api),有了这个免费的可以一个月抠50张的key我们就可以操作了。。。
3.接下来我们先分别定义函数,最后用类搞定它!
3.1初始化,得到api和pic(注意:这里的pic要是你扣的那张图的名字!!)
class ChangePic():
def __init__(self,key,pic):
self.key = key
self.pic = pic
3.2抠出前景图
def remove(self):
"""
单张图片抠图
:return:
"""
rmove = RemoveBg(self.key,'error.log')
path2=''
path = os.getcwd()+'\\'+'picture'+'\\'+ path2.join(self.pic)
print(path)
rmove.remove_background_from_img_file(path)
这里解释下,getcwd()返回的是你当前的工作目录,我的path有些改动的原因是我要处理的原始图在该工作目录的picture文件夹里,而后面的join是因为我发现我的pic是list类型,需要转为str。。。~~~~
3.3更换背景色
这里只在img的那块动了一下,因为open的是str格式。。
3.4上最后一道工序哈哈
if __name__ == '__main__':
hyq=ChangePic('iJL',os.listdir(r"D:\hyq2.0\py\etc\2019-7-09抠图\picture")) #'iJL'放你申请的key
hyq.remove()
hyq.changeBack('A')
4.效果如下:
第一张是抠出的前景图,后一张就是我添的背景了,证件照效果的个人感觉还行吧~~~