百度云人脸识别API人脸库管理

from urllib import request
import base64
import requests
import re
import json
import urllib
import os

# 获取token
def get_token(client_id,client_secret):
    host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id='+ client_id + '&client_secret='+ client_secret + ''
    headers = {'Content-Type': 'application/json; charset=UTF-8'}
    res = requests.post(host, headers=headers)
    access_token = re.findall('"access_token":"(.*?)"', res.text)[0]
    return access_token

# 打开文件夹,进行对比
def geturlPath(path):
    dirs = os.listdir(path)
    lst = []
    for dir1 in dirs:
        pa = path + dir1
        lst.append(pa)
    return lst

# 打开本地图片,并转化为base64
def open_pic2base(image):
    f = open(image,'rb')
    img = base64.b64encode(f.read()).decode('utf-8')
    return img

def search(image3,client_id,client_secret,groups):
    request_url = "https://aip.baidubce.com/rest/2.0/face/v3/search"
    for image in image3:
        a = open_pic2base(image)
        params =json.dumps({"image":a,"image_type":"BASE64","group_id_list":groups})
        access_token = get_token(client_id, client_secret)
        request_url = request_url + "?access_token=" + access_token
        params = params.encode("utf-8")

        request = urllib.request.Request(url=request_url, data=params)

        request.add_header('Content-Type', 'application/json')
        response = urllib.request.urlopen(request)
        content = response.read()
        user = 'not find'
        if content:
            result = json.loads(content.decode('utf-8'))
            print(result)
            if result['error_code']==0:
                lst = []
                sets = {}
                user=result['result']['user_list'][0]
                group_id = user["group_id"]
                user_id=user['user_id']
                score = user['score']
                sets["所在组为"] = group_id
                sets["相似用户为"] = user_id
                sets["相似得分为"] = score
                lst.append(sets)
                print(lst)
                if score > 85:
                    add_face2ku = face_add(image,group_id,user_id)
                else:
                  pass

def face_add(image,group_id,user_id):
    request_url = 'https://aip.baidubce.com/rest/2.0/face/v3/faceset/user/add'
    params = json.dumps({"image":open_pic2base(image),"image_type":"BASE64","group_id":group_id,"user_id":user_id})
    access_token = get_token(client_id, client_secret)
    request_url = request_url + "?access_token=" + access_token
    params = params.encode("utf-8")
    request = urllib.request.Request(url=request_url, data=params)
    request.add_header('Content-Type', 'application/json')
    response = urllib.request.urlopen(request)
    return response

if __name__ == '__main__':
    client_id = '*********************'
    client_secret = '*********************'
    path = r'G:/images5/'

    image3 = geturlPath(path)

    # 将图片与图片库进行对比,传入指定的组
    search(image3, client_id, client_secret, 'actor')