Remove.bg极速精准抠图加PIL上背景色

removebg.py

# 参考-https://www.cnblogs.com/open-yang/p/11240053.html
# API-https://www.remove.bg/

from removebg import RemoveBg
import os

rmbg = RemoveBg("your api", "error.log")  # 前面引号内是你获取的API
path = os.path.join(os.getcwd(), 'images')  # 图片放到程序的同级文件夹 images 里面
# print(os.listdir(path))
for pic in os.listdir(path):
    rmbg.remove_background_from_img_file(f"{path}/{pic}")

---

bg_color.py

from PIL import Image

im = Image.open('images/2.jpg_no_bg.png')  # 加载已经抠图的图片
x, y = im.size
try:
    # (alpha band as paste mask).
    p = Image.new('RGBA', im.size, (255, 0, 0))  # 设置背景颜色为红色
    p.paste(im, (0, 0, x, y), im)
    p.save('images/2_red.png')  # 另存为新的图片
except:
    pass

大家可以修改代码打包成一个命令行工具，然后凭这个完全可以开家淘宝抠图店啦！