通过百度人工智能平台中的人脸检测模块,实现简单的人脸检测,百度人工智能平台免费注册,人脸检测模块免费使用。相关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及其检测结果如下:
检测结果: SUCCESS
图片中包含人脸数: 1
图片中包含人物年龄: 23
图片中包含人物颜值评分: 60.61511993
图片中包含人物性别: female
图片中包含人物种族: yellow
图片中包含人物表情: none
输入图片2及其检测结果如下:
检测结果: SUCCESS
图片中包含人脸数: 1
图片中包含人物年龄: 22
图片中包含人物颜值评分: 77.99679565
图片中包含人物性别: female
图片中包含人物种族: yellow
图片中包含人物表情: smile
输入图片3及其检测结果如下:
检测结果: pic not has face