第一步:安装removebg库
第二步 申请API接口
这里需要用自己邮箱注册,并登陆
https://www.remove.bg/
第三步,代码实现
from removebg import RemoveBg#导入removebg库
import os
rmbg = RemoveBg("VNzVdb7uBAgryt7xkUcDDRXT", "error.log")#替换成自己的API接口
rmbg.remove_background_from_img_file("/Users/郑玉婷/Pictures/Camera Roll/1.jpg")#图片地址,抠完的也会自动保存进去
批量抠图
from removebg import RemoveBg
import os
rmbg = RemoveBg("VNzVdb7uBAgryt7xkUcDDRXT", "error.log")
path = '%s/picture' % os.getcwd()#在源码同级的目录下新建一个图片文件夹放图片
for pic in os.listdir(path):
rmbg.remove_background_from_img_file("%s/%s" % (path, pic))
效果图
单张
批量
总结
Removebg模块的效果还是不错的,用证件照会更好一些,不过由于素材紧缺以及其他原因所以用了别的一些图片,轮廓清晰的效果更好。