[python]百度AI接口实现人像抠图换背景色

最新推荐文章于 2024-09-18 16:59:28 发布

♚暖心~

最新推荐文章于 2024-09-18 16:59:28 发布

阅读量3.5k

点赞数

文章标签： python 百度人工智能

本文链接：https://blog.csdn.net/angel_lf/article/details/121557063

版权

本文介绍了如何利用Python结合百度AI接口实现人像抠图并更换背景色。首先，详细讲解了注册百度AI账号及创建人像分割应用的步骤。接着，展示了代码实现的各个部分，包括库的引入、获取Access Token、核心代码实现等，以及在运行程序前需要注意的事项。最后，作者分享了完成这个功能的喜悦，尽管过程中遇到一些小问题，如图片大小限制和路径处理，但最终成功实现了人像背景替换功能。

摘要由CSDN通过智能技术生成

前言

之前因为业务原因，需要大量制作证件照，在网上找了许多文章，没找到合适的，所以借鉴网上一些资料，自己做了一个证件照背景换色。

最终效果

原图

效果图

一、注册百度AI账号、创建人像分割应用

百度人像分割主页：https://ai.baidu.com/tech/body/seg 按步骤注册，登录，实名认证即可。
在控制台主页找到人体分析

创建应用

里面的需要填写的内容可以随便写，新用户要去领取免费资源，不然使用不了。 在这里插入图片描述 创建完成在应用列表记录 API Key、Secret Key的值 ，稍后要用。至此，注册账号和创建应用的任务就完成了。

二、代码实现

1.引入库

import os
import requests
import base64
import cv2
import numpy as np
from PIL import Image
from pathlib import Path

path = os.getcwd()
paths = list(Path(path).glob('*'))

2.获取Access Token

def get_access_token():
    url = 'https://aip.baidubce.com/oauth/2.0/token'
    data = {
   
        'grant_type': 'client_credentials',  # 固定值
        'client_id': '替换成你的API Key',  # 在开放平台注册后所建应用的API Key
        'client_secret': '替换成你的Secret Key'  # 所建应用的Secret Key
    }
    res = requests.post(url, data=data)
    res = res.json()
    access_token = res['access_token']
    return access_token

3.核心代码

def removebg():
    try:
        request_url = "https://aip.baidubce.com/rest/2.0/image-classify/v1/body_seg"
        # 二进制方式打开图片文件
        f = open(name, 'rb')
        img = base64.b64encode(f.read())
        params = {
   "image":img}
        access_token = get_access_token()
        request_url = request_url + "?access_token=" + access_token
        headers = {
   'content-type': 'application/x-www-form-urlencoded'}
        response = requests.post(request_url, data=params, headers=headers)
        if response