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
大家可以修改代码打包成一个命令行工具,然后凭这个完全可以开家淘宝抠图店啦!