removebg包:一键抠图&换证件照底色

Remove.bg是一个利用AI智能抠图的网站,在这个网站上可以上传图片,然后实现自动抠图并更换背景等,如果不懂编程的可以直接登陆https://www.remove.bg/zh操作

Remove.bg 网站提供了API 接口,可以直接调用并实现抠图。而在python中已经有基于api开发的removebg包,可以非常方便的实现相关功能

首先安装removebg包:

pip install removebg

然后需要一个Remove.bg 网站的API-key.需要登录Remove.bg的官网并注册。注册完成后可以在“我的账户”中查看API-key。复制这个key即可

之后就可以使用remove_background_from_img_file函数快速转化图片的背景了:

from removebg import RemoveBg

rmbg = RemoveBg("YOUR-API-KEY", "error.log")
rmbg.remove_background_from_img_file("testrmbg.jpg")

语法为:
remove_background_from_img_file(img_file_path,size,bg_color),参数如下图所示:

在这里插入图片描述

例如要将证件照背景换为白色底色,并且输出4k分辨率的图片

from removebg import RemoveBg

rmbg = RemoveBg("YOUR-API-KEY", "error.log")
rmbg.remove_background_from_img_file("identificationphoto.jpg",'4k','blank')

同时我们也可以实现照片的批量转换:

from removebg import RemoveBg
import os

rmbg = RemoveBg("YOUR-API-KEY", "error.log")
#os.getcwd() 方法用于返回当前工作目录
#同时将生成的结果放在路径中的picture文件夹中
path='%s/picture'%os.getcwd()
print(path)

for pic in os.listdir(path):  #os.listdir() 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表。
    rmbg.remove_background_from_img_file('%s\%s'%(path,pic))

需要注意的是免费用户系统会自动赠送你一个credit以及50次的免费预览。免费预览就是低像素(最高 25 万像素,例如625 × 400 像素)的抠图,想要获得高像素图像就需要花费credit。所以用完了之后就会拒绝你的api请求.

另外此包还支持修改url图片背景remove_background_from_img_url()和base64图片remove_background_from_base64_img()

最后附上removebg包的github地址:
https://github.com/brilam/remove-bg

官网API文档:
https://www.remove.bg/zh/api

本文首发于博客:http://smilecoc.vip/
微信公众号:Romi的杂货铺