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

最新推荐文章于 2024-07-02 10:13:51 发布

cskywit

最新推荐文章于 2024-07-02 10:13:51 发布

阅读量2.5w

点赞数 6

分类专栏： Python

本文链接：https://blog.csdn.net/cskywit/article/details/81540500

版权

Python 专栏收录该内容

42 篇文章 8 订阅

订阅专栏

通过百度人工智能平台中的人脸检测模块，实现简单的人脸检测，百度人工智能平台免费注册，人脸检测模块免费使用。相关API文档在：http://ai.baidu.com/docs#/Face-Detect-V3/top，依照API文档就可以写出一个简单的检测工具。代码如下：

import base64
import json
import requests

class BaiduPicIndentify:
    def __init__(self,img):
        self.AK = "换成你自己的AK"
        self.SK = "换成你自己的SK"
        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()

输入的图片1.JPG及其检测结果如下：