python利用removebg实现去取背景图,更换背景图

removebg是一个利用AI智能抠图的网站 :  上传图片 – remove.bg
提供了API 接口,可以直接调用并实现抠图,每月有50张免费
使用前需要去官网 https://www.remove.bg/ 注册并获取API KEY

# pip install removebg  安装库函数removebg
# pip install pillow
from removebg import RemoveBg

# rmbg = RemoveBg(r"API KEY", r"D:/error.log")
# rmbg.remove_background_from_img_file(r"D:/mm.png")


# 更换背景颜色
from removebg import RemoveBg
from PIL import Image

# 只支持png格式的图片,jpg格式会报错
# 执行完成之后背景色变成了蓝色 输出图片地址 D:/mm1.png
rmbg = RemoveBg("API KEY​​​​​​​", "error.log")  #主要需要用到API KEY
file_in = "D:/mm.png"
file_out = "D:/mm1.png"

# 换背景色
color = (0, 125, 255)
p, s = file_in.split(".")
rmbg.remove_background_from_img_file(file_in)
file_no_bg = "{}.{}_no_bg.{}".format(p, s, s)
no_bg_image = Image.open(file_no_bg)
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(file_out)

你可能感兴趣的:(Python,java,python,html)