1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193 | ##APPID、APISecret、APIKey一定要填写！！！##
import tkinter as tk
from tkinter import filedialog
from tkinter import Label
from tkinter import ttk
from PIL import Image, ImageTk
import json
import requests
import base64
import hmac
import hashlib
from datetime import datetime
from time import mktime
from wsgiref.handlers import format_date_time
from urllib.parse import urlencode
class AssembleHeaderException(Exception):
def __init__(self, msg):
self.message = msg
class Url:
def __init__(this, host, path, schema):
this.host = host
this.path = path
this.schema = schema
def sha256base64(data):
sha256 = hashlib.sha256()
sha256.update(data)
digest = base64.b64encode(sha256.digest()).decode(encoding='utf-8')
return digest
def parse_url(request_url):
stidx = request_url.index("://")
host = request_url[stidx + 3:]
schema = request_url[:stidx + 3]
edidx = host.index("/")
if edidx <= 0:
raise AssembleHeaderException("invalid request url:" + request_url)
path = host[edidx:]
host = host[:edidx]
u = Url(host, path, schema)
return u
def assemble_ws_auth_url(request_url, method="GET", api_key="", api_secret=""):
u = parse_url(request_url)
host = u.host
path = u.path
now = datetime.now()
date = format_date_time(mktime(now.timetuple()))
signature_origin = "host: {}\ndate: {}\n{} {} HTTP/1.1".format(host, date, method, path)
signature_sha = hmac.new(api_secret.encode('utf-8'), signature_origin.encode('utf-8'),
digestmod=hashlib.sha256).digest()
signature_sha = base64.b64encode(signature_sha).decode(encoding='utf-8')
authorization_origin = "api_key=\"%s\", algorithm=\"%s\", headers=\"%s\", signature=\"%s\"" % (
api_key, "hmac-sha256", "host date request-line", signature_sha)
authorization = base64.b64encode(authorization_origin.encode('utf-8')).decode(encoding='utf-8')
values = {
"host": host,
"date": date,
"authorization": authorization
}
return request_url + "?" + urlencode(values)
def gen_body(appid, img1_path, img2_path, server_id):
with open(img1_path, 'rb') as f:
img1_data = f.read()
with open(img2_path, 'rb') as f:
img2_data = f.read()
body = {
"header": {
"app_id": appid,
"status": 3
},
"parameter": {
server_id: {
"service_kind": "face_compare",
"face_compare_result": {
"encoding": "utf8",
"compress": "raw",
"format": "json"
}
}
},
"payload": {
"input1": {
"encoding": "jpg",
"status": 3,
"image": str(base64.b64encode(img1_data), 'utf-8')
},
"input2": {
"encoding": "jpg",
"status": 3,
"image": str(base64.b64encode(img2_data), 'utf-8')
}
}
}
return json.dumps(body)
def run(appid, apikey, apisecret, img1_path, img2_path, server_id='s67c9c78c'):
url = 'http://api.xf-yun.com/v1/private/{}'.format(server_id)
request_url = assemble_ws_auth_url(url, "POST", apikey, apisecret)
headers = {'content-type': "application/json", 'host': 'api.xf-yun.com', 'app_id': appid}
response = requests.post(request_url, data=gen_body(appid, img1_path, img2_path, server_id), headers=headers)
resp_data = json.loads(response.content.decode('utf-8'))
result = base64.b64decode(resp_data['payload']['face_compare_result']['text']).decode()
return result
def browse_file(entry_widget, image_label, img_num):
file_path = filedialog.askopenfilename()
if file_path:
entry_widget.delete(0, tk.END)
entry_widget.insert(0, file_path)
load_and_display_image(file_path, image_label, img_num)
def load_and_display_image(file_path, image_label, image_num):
try:
image = Image.open(file_path)
image = image.resize((200, 200))
photo = ImageTk.PhotoImage(image)
image_label.config(image=photo)
image_label.image = photo
if image_num == 1:
global img1_data
img1_data = image
elif image_num == 2:
global img2_data
img2_data = image
except Exception as e:
result_label.config(text="加载图片出错，请检查文件格式")
def compare_faces():
if 'img1_data' not in globals() or 'img2_data' not in globals():
result_label.config(text="请选择两张图片进行比对")
return
try:
img1_path = 'img1.jpg'
img2_path = 'img2.jpg'
img1_data.save(img1_path)
img2_data.save(img2_path)
result = run(appid='',#自行申请填写
apisecret='',#自行申请填写
apikey='',#自行申请填写
img1_path=img1_path,
img2_path=img2_path)
score = float(json.loads(result)['score'])
if score >= 0.67:
result_label.config(text=f"这两张图片是同一个人，相似度：{score:.2f}")
else:
result_label.config(text=f"这两张图片不是同一个人，相似度：{score:.2f}")
except Exception as e:
result_label.config(text="比对出错，请检查图片和配置")
root = tk.Tk()
root.title("人脸比对")
# 设置样式主题
style = ttk.Style()
style.configure('TButton', font=('Helvetica', 12))
style.configure('TLabel', font=('Helvetica', 14))
frame = ttk.LabelFrame(root, text="选择图片")
frame.grid(row=0, column=0, columnspan=2, padx=10, pady=10, sticky="ew")
entry1 = tk.Entry(frame, width=50)
entry2 = tk.Entry(frame, width=50)
entry1.grid(row=0, column=0, padx=10, pady=10)
entry2.grid(row=1, column=0, padx=10, pady=10)
separator = ttk.Separator(root, orient='horizontal')
separator.grid(row=1, column=0, columnspan=2, sticky="ew")
button1 = ttk.Button(root, text="选择图片1", command=lambda: browse_file(entry1, img_label1, 1))
button2 = ttk.Button(root, text="选择图片2", command=lambda: browse_file(entry2, img_label2, 2))
button1.grid(row=2, column=0, padx=10, pady=10)
button2.grid(row=2, column=1, padx=10, pady=10)
compare_button = ttk.Button(root, text="比对图片", command=compare_faces)
compare_button.grid(row=3, column=0, columnspan=2, pady=20)
result_label = Label(root, text="", font=("Helvetica", 14))
result_label.grid(row=4, column=0, columnspan=2, padx=10, pady=10)
img_label1 = tk.Label(root)
img_label1.grid(row=5, column=0, padx=10, pady=10)
img_label2 = tk.Label(root)
img_label2.grid(row=5, column=1, padx=10, pady=10)
root.mainloop()
|
