【基于百度AI的人脸识别Python实现】

调用百度AI的baidu-aip库进行人脸识别，baidu-aip库可以使用pip install baidu-aip进行安装

关于人脸检测请求参数可参考传送门

运行代码如下：

# coding=UTF-8
# 利用baidu-aip库进行人脸识别
import cv2
import matplotlib.pyplot as plt
from aip import AipFace


def detection(APP_ID, API_KEY, SECRET_KEY, filename, maxnum):
    '''

    :param APP_ID: https://console.bce.baidu.com/ai/创建人脸检测应用对应的APP_ID
    :param API_KEY: https://console.bce.baidu.com/ai/创建人脸检测应用对应的API_KEY
    :param SECRET_ID: https://console.bce.baidu.com/ai/创建人脸检测应用对应的SECRET_ID
    :param filename: 图片路径
    :param maxnum: 最大检测数
    :return:
    '''
    # 初始化AirFace对象
    aipface = AipFace(APP_ID, API_KEY, SECRET_KEY)

    # 设置
    options = {
        'max_face_num': 10,  # 检测人脸的最大数量
        'face_fields': "age,beauty,expression,faceshape",
    }

    # 读取文件内容
    def get_file_content(filePath):
        with open(filePath, 'rb') as fp:
            return fp.read()

    result = aipface.detect(get_file_content(filename), options)
    return result


def result_show(filename, result):
    '''

    :param filename: 原始图像
    :param result: 检测结果
    :return:
    '''
    img = cv2.imread(filename)
    face_num = len(result['result'])
    for i in range(face_num):
        location = result['result'][i]['location']
        left_top = (location['left'], location['top'])
        right_bottom = (left_top[0] + location['width'], left_top[1] + location['height'])
        cv2.rectangle(img, left_top, right_bottom, (200, 100, 0), 2)

    cv2.imshow('img', img)
    cv2.waitKey(0)


if __name__ =='__main__':

    # 定义APP_ID、API_KEY、SECRET_KEY
    APP_ID = '10761235'
    API_KEY = 'PF4Ytr53mYdeKY8fjChE4Elh'
    SECRET_KEY = 'q83srw3x3YDg6nTIh4I7ZzscntT7qWWO '

    filename = 'girl.jpg'
    result = detection(APP_ID, API_KEY, SECRET_KEY, filename, 10)
    result_show(filename, result)

运行结果如下：

换张图像试试效果：