调用百度人脸识别API实现简单的颜值检测

Posted 2021-07-11 ZHW_AI课题组

目录

• 作者介绍
• 注册百度智能云账号
• 代码实现
• 运行结果

作者介绍

乔冠华，女，西安工程大学电子信息学院，2020级硕士研究生，张宏伟人工智能课题组。
研究方向：机器视觉与人工智能。
电子邮件：[e-mail]

注册百度智能云账号

进入：百度智能云并创建应用获取AppID,API Key,Secret Key

免费注册登录百度云账号进入下图界面，选择“产品服务”中“人脸识别”模块
点击“创建应用”，免费创建名为“人脸颜值检测”的个人应用模块，创建成功的应用模块如下，并从中获取个人应用模块的API Key和Secret Key。：

代码实现

在Pycharm中输入代码，并更改对应位置个人账号信息（API Key、Secret Key），运行程序后，输入待测试人脸图像路径，实现人脸颜值检测。

import base64
import json
import requests
 
class BaiduPicIndentify:
    def __init__(self,img):
        self.AK = "换成个人的API Key"
        self.SK = "换成个人的Secret Key"
        self.img_src = img
        self.headers = {
            "Content-Type": "application/json; charset=UTF-8"
        }
 
    def get_accessToken(self):
        host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=' + self.AK + '&client_secret=' + self.SK
        response = requests.get(host, headers=self.headers)
        json_result = json.loads(response.text)
        return json_result['access_token']
 
    def img_to_BASE64(slef,path):
        with open(path,'rb') as f:
            base64_data = base64.b64encode(f.read())
            return base64_data
 
    def detect_face(self):
        # 人脸检测与属性分析
        img_BASE64 = self.img_to_BASE64(self.img_src)
        request_url = "https://aip.baidubce.com/rest/2.0/face/v3/detect"
        post_data = {
            "image": img_BASE64,
            "image_type": "BASE64",
            "face_field": "gender,age,beauty,gender,race,expression",
            "face_type": "LIVE"
        }
        access_token = self.get_accessToken()
        request_url = request_url + "?access_token=" + access_token
        response = requests.post(url=request_url, data=post_data, headers=self.headers)
        json_result = json.loads(response.text)
        if json_result['error_msg']!='pic not has face':
            print("图片中包含人脸数：", json_result['result']['face_num'])
            print("图片中包含人物年龄：", json_result['result']['face_list'][0]['age'])
            print("图片中包含人物颜值评分：", json_result['result']['face_list'][0]['beauty'])
            print("图片中包含人物性别：", json_result['result']['face_list'][0]['gender']['type'])
            print("图片中包含人物种族：", json_result['result']['face_list'][0]['race']['type'])
            print("图片中包含人物表情：", json_result['result']['face_list'][0]['expression']['type'])
 
if __name__=='__main__':
    img_src=input('请输入需要检测的本地图片路径:')
    baiduDetect = BaiduPicIndentify(img_src)
    baiduDetect.detect_face()

运行结果

参考链接：
https://www.jb51.net/article/207785.htm
https://blog.csdn.net/cskywit/article/details/81540500