【Python案例】用某度AI接口实现抠图并改图片底色

最新推荐文章于 2024-04-22 10:11:59 发布

「已注销」

最新推荐文章于 2024-04-22 10:11:59 发布

阅读量1.2w

点赞数 29

分类专栏： python案例 pytho基础文章标签： python 人工智能开发语言

本文链接：https://blog.csdn.net/aliYz/article/details/126932847

版权

本文介绍如何使用Python结合百度AI接口进行人像抠图并更改图片背景色。通过注册百度AI账号，获取API Key和Secret Key，然后编写Python代码，包括获取Access Token、图片处理等步骤，实现方便快捷的图片处理功能。

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

前言

嗨嗨，大家好呀，我是小圆。有时候我们的证件照需要换底色，又来不及去照相馆拍照，用ps也不好抠图，所以今天给你们分享一下如何用python来抠图，并换底色

请添加图片描述

正文

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

对文章有问题可以私信我或者来这里哦 https://jq.qq.com/?_wv=1027&k=s5bZE0K3

百度人像分割主页：按步骤注册，登录，实名认证即可。
在控制台主页找到人体分析

在这里插入图片描述

创建应用

在这里插入图片描述

里面的需要填写的内容可以随便写，新用户要去领取免费资源，不然使用不了。

在这里插入图片描述

创建完成在应用列表记录 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

核心代码

对文章有问题可以私信我或者来这里哦 https://jq.qq.com/?_wv=1027&k=s5bZE0K3

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:
            res = response.json()