python人脸对比

Posted 2020-11-03 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)