1.安装
removebg和PIL模块
运行命令提示行,执行命令:
pip install removebg
pip install Pillow
2.获取API Key
搜索removebg,进入官网,注册账号后,进入工具和API——API Docs——Get API Key,即可获取API密钥。
3.python抠图和填充
抠图,获取无背景图片aaa.jpg_no_bg.png
from removebg import RemoveBg
rmbg=RemoveBg("**********","error.log") #AI密钥
rmbg.remove_background_from_img_file("C:/Users/Desktop/aaa.jpg") #输入图片,jpg格式,试过bmp未成功
填充背景颜色
# 给去除了背景的图像添加背景颜色
from PIL import Image
# 输入已经去除背景的图像
im = Image.open('C:/Users/Desktop/aaa.jpg_no_bg.png')
x, y = im.size
try:
# 填充红色背景255,0,0 白色是255,255,255
p = Image.new('RGBA', im.size, (255, 0, 0))
p.paste(im, (0, 0, x, y), im)
# 保存填充后的图片
p.save('C:/Users/Desktop/aaa.jpg_red_bg.png')
except:
with open('./error.log', 'a') as f:
f.write('background change fail .')
成功更换照片底色
参考:
https://blog.csdn.net/Chad97/article/details/100010237
https://blog.csdn.net/holly_z_p_f/article/details/100565065