python人脸对比
import sys
import ssl
from urllib import request,parse
# client_id 为官网获取的AK, client_secret 为官网获取的SK
#获取token
def get_token():
client_id =API Key
client_secret =Secret Key
host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=%s&client_secret=%s'%(client_id,client_secret)
req = request.Request(host)
req.add_header('Content-Type', 'application/json; charset=UTF-8')
response = request.urlopen(req)
#获得请求结果
content = response.read()
#结果转化为字符
content = bytes.decode(content)
#转化为字典
content = eval(content[:-1])
return content['access_token']
#转换图片
#读取文件内容,转换为base64编码
#二进制方式打开图文件
def imgdata(file1path,file2path):
import base64
f=open(r'%s' % file1path,'rb')
pic1=base64.b64encode(f.read())
f.close()
f=open(r'%s' % file2path,'rb')
pic2=base64.b64encode(f.read())
f.close()
#将图片信息格式化为可提交信息,这里需要注意str参数设置
params = {"images":str(pic1,'utf-8') + ',' + str(pic2,'utf-8')}
return params
#提交进行对比获得结果
def img(file1path,file2path):
token = get_token()
#人脸识别API
#url = 'https://aip.baidubce.com/rest/2.0/face/v2/detect?access_token='+token
#人脸对比API
url = 'https://aip.baidubce.com/rest/2.0/face/v2/match?access_token='+token
params = imgdata(file1path,file2path)
#urlencode处理需提交的数据
data = parse.urlencode(params).encode('utf-8')
req = request.Request(url,data=data)
req.add_header('Content-Type', 'application/x-www-form-urlencoded')
response = request.urlopen(req)
content = response.read()
content = bytes.decode(content)
content = eval(content)
print(content)
#获得分数
score = content['result'][0]['score']
if score>80:
return '照片相似度:'+str(score)+',同一个人'
else:
return '照片相似度:'+str(score)+',不是同一个人'
if __name__ == '__main__':
file1path = 'd:/ym1.jpg'
file2path = 'd:/ym2.jpg'
res = img(file1path,file2path)
print(res)