Spaces:
Runtime error
Runtime error
File size: 1,294 Bytes
16eecd4 0cba43f 16eecd4 |
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 |
from utils.database import get_database
import os
import face_recognition
import cv2
import numpy as np
from PIL import Image
def recognize(image,tolerance):
database = get_databse()
known_encoding = [database[id]['encoding'] for id in database.keys()]
name = 'Unknown'
face_id = 'Unknown'
face_locations = face_recognition.face_locations(image)
face_encodings = face_recognition.face_encodings(image,face_locations)
for (top,right,bottom,left), face_encoding in zip(face_locations,face_encodings):
matches = face_recognition.compare_faces(known_encoding,face_encoding,tolerance=tolerance)
distance = face_recognition.face_distance(known_encoding,face_encoding)
name = 'Unknown'
face_id = 'Unknown'
if True in matches:
match_index = matches.index(True)
name = database[match_index]['name']
face_id = database[match_index]['face_id'].split("_")[1]
distance = round(distance[match_index],2)
cv2.putText(image,str(distance),(left,top-30),cv2.FONT_HERSHEY_SIMPLEX,0.75,(0,255,0),2)
cv2.rectangle(image,(left,top),(right,bottom),(0,255,0),2)
cv2.putText(image,name,(left,top-10),cv2.FONT_HERSHEY_SIMPLEX,0.75,(0,255,0),2)
return image, name, face_id
|