38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
from django.conf import settings
|
|
import uuid
|
|
import face_recognition
|
|
import os
|
|
from apps.hrm.models import Employee
|
|
from apps.system.models import User
|
|
|
|
class HRMService:
|
|
|
|
@classmethod
|
|
def face_compare_from_base64(cls, base64_data):
|
|
filename = str(uuid.uuid4())
|
|
filepath = settings.BASE_DIR +'/temp/' + filename +'.png'
|
|
with open(filepath, 'wb') as f:
|
|
f.write(base64_data)
|
|
try:
|
|
unknown_picture = face_recognition.load_image_file(filepath)
|
|
unknown_face_encoding = face_recognition.face_encodings(unknown_picture)[0]
|
|
os.remove(filepath)
|
|
except:
|
|
os.remove(filepath)
|
|
return None, '头像解码失败'
|
|
|
|
# 匹配人脸库
|
|
user_faces = Employee.objects.filter(face_data__isnull=False, user__is_active=True).values('user', 'face_data')
|
|
user_l = []
|
|
face_l = []
|
|
for i in user_faces:
|
|
user_l.append(i['user'])
|
|
face_l.append(i['face_data'])
|
|
|
|
results = face_recognition.compare_faces(face_l, unknown_face_encoding, tolerance=0.5)
|
|
for index, value in enumerate(results):
|
|
if value:
|
|
# 识别成功
|
|
user = User.objects.get(id=user_l[index])
|
|
return user, ''
|
|
return None, '识别失败' |