本文目录：前期准备-编写代码-测试结果-文末福利

本文作者：小赖同学

不知不觉就到了2月中旬了，因为肺炎疫情大家都一再延迟了上班上学，天天睡到大中午，起来吃个小午餐，玩手机睡午觉又到大晚上，哪都不敢去，被迫放假是真的很难受。

而我，虽然已经上班了，但下班了哪都不敢去，超级无聊。这几天，在整理以前的摄影照片时无意中发现了几张证件照，又看到了电脑桌面的PS，突然间就来了点灵感，好像好像又不那么无聊了。

我可以用Python来抠图吗？可以给证件照换个底色吗？我可以负责任地告诉你，经过几天实践得知，可以！只有几行代码而且还很快！(提前偷偷告诉你，文末还有福利哦)

啥，想看看效果图？好，上动图。

啥，想看看是如何实现的？好，接下来请听我慢慢叙述。

前期准备

1、工具挑选

Python要实现抠图功能，网上有很多方法。我这边经过对比，挑选了RemoveBg这款工具。RemoveBg是国外的一个专业为人物去除背景的神器，它利用的是智能AI抠图，100%全自动化，全程无需人为操作抠图，可以说是非常简单快捷的一款工具了，并且号称只要5秒就能完成去除背景，这个工具目前完全免费，提供PNG图片下载。

官网支持网上在线抠图，只需要简单注册获取其API KEY就可以使用，每个API KEY每月免费50张，但由于是国外服务器，使用过程会感觉页面加载有点缓慢，导出的图片大小也会有所压缩和限制。

最重要的是Python支持RemoveBg，只需要下载安装这个模块就可以使用了，完美符合此文的主题。

2、获取API KEY

搜索RemoveBg官网，进入官网注册一个账号就可以获得API KEY。这个API KEY是相当于一个“身份ID”，实现证件照换底色需要导入第三方模块removebg，有了这个ID就可以使用到RemoveBg的“技术支持”。

RemoveBg官网：https://www.remove.bg/zh，API KEY获取网址：https://www.remove.bg/zh/api，获取API KEY的具体操作也很简单：

编写代码

1、安装removebg模块和Pillow模块

Windows + R打开运行窗口，然后输入cmd打开命令提示符窗口，在命令行终端输入pip install removebg进行removebg模块的安装，输入pip install Pillow进行Pillow模块的安装。

稍等一会，提示安装成功就可以使用这两个模块了。(备注：我这是因为配置好了python环境的，没安装python的可以自己搜索下如何安装。)

2、代码介绍

安装好了需要的工具模块后，就可以开始编写代码了。证件照换底色使用的主要代码也就十行左右。

rmbg = RemoveBg(api_key, "error.log")
rmbg.remove_background_from_img_file(old_image_path)

首先是获得RemoveBg，传入上述所说的API KEY，再将需要更换底色的证件照路径传入其remove_background_from_img_file()方法，此方法会在证件照路径下新生成一个抠除底色的图像。

no_bg_image = Image.open(no_bg_image_path)
x, y = no_bg_image.size
try:
    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(new_image_path)
except:
    print("image matting except")

接着通过PIL下的Image打开抠除底色的图像，获取到其分辨率大小等数据后再通过Image新生成一个特定底色的证件照，color即是需要更换成的底色，color可以是"red"等英文字符串，也可以是(255, 0, 0)等RGB值。最后把更换底色后的证件照保存到你选择的路径下，底色更换就完成了。

3、完整代码

通过上面第二点的介绍，就已基本清楚了证件照更换底色的代码实现原理，最后就附上完整的简单整合的实现代码。

import os

from PIL import Image
from removebg import RemoveBg

def image_matting(old_image_path, new_image_path, api_key, color):
    # API KEY获取官方网站：https://www.remove.bg/zh/api
    rmbg = RemoveBg(api_key, "error.log")
    rmbg.remove_background_from_img_file(old_image_path)
    
    parent_path = os.path.dirname(old_image_path)
    old_image_name = os.path.split(old_image_path)[-1]
    no_bg_image_name = old_image_name + "_no_bg.png"
    if no_bg_image_name in os.listdir(parent_path):
        no_bg_image_path = parent_path + "/" + no_bg_image_name
        no_bg_image = Image.open(no_bg_image_path)
        x, y = no_bg_image.size
        try:
          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(new_image_path)
        except:
          print("image matting except")
    else:
        print("image matting fail")