Beispiel #1
0
import sys

from face_detector import FaceDetector
from face_recognizer import FaceRecognizer
import psycopg2

fd = FaceDetector()
fr = FaceRecognizer()

faces = fd.detect_file("test2.jpg")
users = []
for face in faces:
    user_id, confidence = fr.recognize(face,
                                       "f5e053ef-4f1d-4586-87e5-449864754549")
    try:
        conn = psycopg2.connect(
            "dbname='InfiniteDrop' user='******' host='127.0.0.1' password='******'"
        )
    except:
        print("unable to connect to the database")
        sys.exit(0)
    cur = conn.cursor()
    cur.execute("""SELECT name from face_user where id=%s""", [user_id])
    result = cur.fetchone()
    users.append(result[0])

print(users)