1 importbase642 importurllib3 importurllib.request4 importurllib.parse5 importtime6 importjson7
8 #所需的图片要以base64编码的形式上传二进制字符串
9 defimage_to_base64(image_path):10 image_data = open(image_path, "rb")11 temp =image_data.read()12 base64_data =base64.b64encode(temp)13 image_data.close()14 data_string =str(base64_data)15 data_string = data_string.strip('\'b')16 returndata_string17
18 defcompare(image1, image2):19 url = 'http://apis.baidu.com/idl_baidu/faceverifyservice/face_compare'
20 reg = r'{"params": [{"cmdid": "1000", "appid": "你的apikey", "clientip": "你的ip","type": "st_groupverify", "groupid": "12345", "versionnum": "1.0.0.1", "usernames": {"name1": "name1", "name2": "name2"}, "images": {"name1": "%s", "name2": "%s"}, "cates": {"name1":"7", "name2":"7"}}], "jsonrpc": "2.0","method": "Compare"}' %(image1, image2)21 byte_data = reg.encode(encoding="utf-8")22 req = urllib.request.Request(url, data=byte_data)23 req.add_header("apikey", "你的apikey")24 #请求第一次可能会出错,重新请求一次就好了
25 try:26 resp =urllib.request.urlopen(req)27 content =resp.read()28 ifcontent:29 print(content)30 returncontent31 exceptException:32 resp =urllib.request.urlopen(req)33 content =resp.read()34 ifcontent:35 print(content)36 returncontent37
38 defcompare_test():39 lfw_file = open("pairs.txt")40 res_file = open("res_baidu.txt", "a+")41 count =042 while 1:43 image_path = "你本地的lfw图片地址"
44 line =lfw_file.readline()45 if notline:46 break
47 line = line.strip('\n')48 images = line.split('\t')49 if len(images) > 3:50 register_image = image_path + images[0] + "\\" + images[0] + "_"
51 if len(images[1]) < 2:52 register_image = register_image + "000" + images[1] + ".jpg"
53 elif len(images[1]) < 3:54 register_image = register_image + "00" + images[1] + ".jpg"
55 elif len(images[1]) < 4:56 register_image = register_image + "0" + images[1] + ".jpg"
57 else:58 register_image = register_image + images[1] + ".jpg"
59
60 verify_image = image_path + images[2] + "\\" + images[2] + "_"
61 if len(images[3]) < 2:62 verify_image = verify_image + "000" + images[3] + ".jpg"
63 elif len(images[3]) < 3:64 verify_image = verify_image + "00" + images[3] + ".jpg"
65 elif len(images[3]) < 4:66 verify_image = verify_image + "0" + images[3] + ".jpg"
67 else:68 verify_image = verify_image + images[3] + ".jpg"
69 else:70 register_image = image_path + images[0] + "\\" + images[0] + "_"
71 if len(images[1]) < 2:72 register_image = register_image + "000" + images[1] + ".jpg"
73 elif len(images[1]) < 3:74 register_image = register_image + "00" + images[1] + ".jpg"
75 elif len(images[1]) < 4:76 register_image = register_image + "0" + images[1] + ".jpg"
77 else:78 register_image = register_image + images[1] + ".jpg"
79
80 verify_image = image_path + images[0] + "\\" + images[0] + "_"
81 if len(images[2]) < 2:82 verify_image = verify_image + "000" + images[2] + ".jpg"
83 elif len(images[2]) < 3:84 verify_image = verify_image + "00" + images[2] + ".jpg"
85 elif len(images[2]) < 4:86 verify_image = verify_image + "0" + images[2] + ".jpg"
87 else:88 verify_image = verify_image + images[2] + ".jpg"
89
90 content =compare(image_to_base64(register_image), image_to_base64(verify_image))91 content =str(content)92 content = content.strip('b\'')93 content =json.loads(content)94
95 if (int(count/300)) % 2 ==0:96 flag = 1
97 else:98 flag =099
100 if content['result']['_ret']['reslist']:101 score = content['result']['_ret']['reslist']['name2|name1']102 else:103 score = "-1"
104
105 res_file.write(line + '\t' + score + '\t' + str(flag) + '\n')106 count += 1
107
108 lfw_file.close()109 res_file.close()110
111 compare_test()