人脸识别登录实战5:摄像头识别多人脸


APP

零度逍遥

2019-06-21 17:03:18

能让摄像头识别出你自己的脸还不够,因为你不带朋友们玩,朋友们会生气的,所以你需要一个可以检测多个人脸的程序。

首先还是一样,找到你和你的朋友们的人脸图片,将它们放在一个文件夹里,你的朋友是lisi和wangwu吗?不是?没关系,在这里他们就是。

然后复制粘贴以下Python代码:

# -*- coding: utf-8 -*-
# 摄像头头像识别
import face_recognition
import cv2

video_capture = cv2.VideoCapture(0)

# 本地图像
image1 = face_recognition.load_image_file("zhangsan.jpg")
face_encoding1 = face_recognition.face_encodings(image1)[0]

# 本地图像二
image2 = face_recognition.load_image_file("lisi.jpg")
face_encoding2 = face_recognition.face_encodings(image2)[0]

# 本地图片三
image3 = face_recognition.load_image_file("wangwu.jpg")
face_encoding3 = face_recognition.face_encodings(image3)[0]

# Create arrays of known face encodings and their names
# 脸部特征数据的集合
known_face_encodings = [
    face_encoding1,
    face_encoding2,
    face_encoding3
]

# 人物名称的集合
known_face_names = [
    "zhangsan",
    "lisi",
    "wangwu"
]

face_locations = []
face_encodings = []
face_names = []
process_this_frame = True

while True:
    # 读取摄像头画面
    ret, frame = video_capture.read()

    # 改变摄像头图像的大小,图像小,所做的计算就少
    small_frame = cv2.resize(frame, (0, 0), fx=0.25, fy=0.25)

    # opencv的图像是BGR格式的,而我们需要是的RGB格式的,因此需要进行一个转换。
    rgb_small_frame = small_frame[:, :, ::-1]

    # Only process every other frame of video to save time
    if process_this_frame:
        # 根据encoding来判断是不是同一个人,是就输出true,不是为flase
        face_locations = face_recognition.face_locations(rgb_small_frame)
        face_encodings = face_recognition.face_encodings(rgb_small_frame, face_locations)

        face_names = []
        for face_encoding in face_encodings:
            # 默认为unknown
            matches = face_recognition.compare_faces(known_face_encodings, face_encoding)
            name = "Unknown"

            # if match[0]:
            #     name = "michong"
            # If a match was found in known_face_encodings, just use the first one.
            if True in matches:
                first_match_index = matches.index(True)
                name = known_face_names[first_match_index]
            face_names.append(name)

    process_this_frame = not process_this_frame

    # 将捕捉到的人脸显示出来
    for (top, right, bottom, left), name in zip(face_locations, face_names):
        # Scale back up face locations since the frame we detected in was scaled to 1/4 size
        top *= 4
        right *= 4
        bottom *= 4
        left *= 4

        # 矩形框
        cv2.rectangle(frame, (left, top), (right, bottom), (0, 0, 255), 2)

        #加上标签
        cv2.rectangle(frame, (left, bottom - 35), (right, bottom), (0, 0, 255), cv2.FILLED)
        font = cv2.FONT_HERSHEY_DUPLEX
        cv2.putText(frame, name, (left + 6, bottom - 6), font, 1.0, (255, 255, 255), 1)

    # Display
    cv2.imshow('monitor', frame)

    # 按Q退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

video_capture.release()
cv2.destroyAllWindows()

运行python文件。

现在把你的朋友们都叫到摄像头跟前吧,不管你是笑还是哭,但记住都要面对摄像头。

什么?还没有显示出姓名,把你们的脸蛋动动,凑近一点看看呢!

bingo!你们的脸上面显示出了对应的图片文件名,当然这可能会是你们的姓名。