欢迎访问宙启技术站
智能推送

使用Python编写一个简单的人脸识别应用

发布时间:2023-12-04 14:05:06

人脸识别是一种将图像或视频中的人脸与数据库中的已知人脸进行比对的技术。Python提供了多个库和工具来实现人脸识别的功能,其中最常用的是dlib和OpenCV。

一、安装依赖库

要使用Python进行人脸识别,首先需要安装dlib和OpenCV库。可以通过pip命令来进行安装。

pip install dlib

pip install opencv-python

二、人脸检测

人脸识别的第一步是检测图像或视频帧中的人脸。使用dlib库提供的人脸检测器可以很方便地实现人脸检测功能。

import dlib

import cv2

detector = dlib.get_frontal_face_detector()

def detect_faces(image):

    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

    faces = detector(gray)

    return faces

image = cv2.imread("image.jpg")

faces = detect_faces(image)

for face in faces:

    x, y, w, h = face.left(), face.top(), face.width(), face.height()

    cv2.rectangle(image, (x, y), (x+w, y+h), (0, 255, 0), 2)

cv2.imshow("Faces", image)

cv2.waitKey(0)

cv2.destroyAllWindows()

上述代码中,首先导入了dlib和cv2库,并创建了一个人脸检测器对象detector。然后定义了一个detect_faces函数,该函数接受一张图像作为输入,将其转换为灰度图像,并调用detector进行人脸检测,返回检测到的人脸区域。接下来,读取一张图像,调用detect_faces函数进行人脸检测,并将检测到的人脸区域用矩形框标记出来,最后展示结果图像。

三、人脸识别

人脸识别的下一步是将检测到的人脸与已知人脸进行比对。可以使用dlib的人脸识别器来实现这个功能。

import dlib

import cv2

# 加载已知人脸的特征向量

known_face_encodings = [

    #[已知人脸1的特征向量],

    #[已知人脸2的特征向量],

    #...

]

# 加载已知人脸的标签

known_face_labels = [

    #已知人脸1的标签,

    #已知人脸2的标签,

    #...

]

# 创建人脸识别器对象

face_recognizer = dlib.face_recognition_model_v1("shape_predictor_68_face_landmarks.dat")

def recognize_faces(image, faces):

    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

    face_landmarks = [face_recognizer(gray, face) for face in faces]

    face_encodings = [dlib.vector_to_numpy(face_descriptor) for face_descriptor in face_landmarks]

    face_labels = []

    for face_encoding in face_encodings:

        # 比对人脸特征向量和已知人脸的特征向量

        matches = dlib.face_distance(known_face_encodings, face_encoding)

        # 找到最小距离对应的已知人脸

        min_distance_index = np.argmin(matches)

        min_distance = matches[min_distance_index]

        # 判断是否为已知人脸

        if min_distance < 0.6:

            face_label = known_face_labels[min_distance_index]

        else:

            face_label = "Unknown"

        face_labels.append(face_label)

    return face_labels

image = cv2.imread("image.jpg")

faces = detect_faces(image)

face_labels = recognize_faces(image, faces)

for face, face_label in zip(faces, face_labels):

    x, y, w, h = face.left(), face.top(), face.width(), face.height()

    cv2.rectangle(image, (x, y), (x+w, y+h), (0, 255, 0), 2)

    cv2.putText(image, face_label, (x, y-10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 255, 0), 2)

cv2.imshow("Faces", image)

cv2.waitKey(0)

cv2.destroyAllWindows()

上述代码首先加载已知人脸的特征向量和标签。然后创建了一个人脸识别器对象face_recognizer,该对象使用预训练模型shape_predictor_68_face_landmarks.dat。接下来定义了一个recognize_faces函数,该函数接受一张图像和检测到的人脸区域作为输入参数,将图像转换为灰度图像,并使用face_recognizer生成人脸特征向量。然后将人脸特征向量与已知人脸的特征向量进行比对,找到最小距离对应的已知人脸,并判断是否为已知人脸。最后读取一张图像,调用detect_faces和recognize_faces函数,并将识别结果标记在图像上。

以上就是使用Python编写一个简单的人脸识别应用的例子。这只是一个简单的示例,实际应用中可能需要更多的功能和完善的算法来提高人脸识别的准确性和性能。