python 人脸识别

这次用python自带的opencv以及face_recognition库实现简单的人脸识别，用pyqt5做了个简单的界面，将程序分为图片人脸识别以及实时人脸识别。
码源：https://github.com/big-indoorsman/face_recognition

一，准备工作
这里我所用的是pycharm2018，python3.6.7.
主要安装的库：PyQt5,face_recognition,cv2

pip install opencv-python#安装cv2

若安装face_recognition失败，请参考这里
文件结构如下：

二，代码实现
store.py

import face_recognition
import pickle
import csv
#打开存储信息表的csv文件，也可以直接存到数据库中
with open('picture.csv','r') as f:
    reader = csv.reader(f)
    header_row = next(reader)
    name,path= [], []
    for row in reader:
        name.append(row[0])
        path.append(row[1])
known_face_encodings=[]
for i in path:
    #根据路径打开图片，信息表里的路径不能包含中文，否则无法读取
    img = face_recognition.load_image_file(i)
    #将图片中的人脸特征数据转换成数组
    locations = face_recognition.face_locations(img)
    face_encodings = face_recognition.face_encodings(img, locations)[0]#假定每张图片只有一张脸以及对应的姓名，如果超出一个人，则取第一组数据
    known_face_encodings.append(face_encodings)
known_names=name
#将名字与人脸特征数据存储到pickle文件中
file = open('show_name.pkl', 'wb')
pickle.dump(known_names, file)
file1 = open('show_data.pkl', 'wb')
pickle.dump(known_face_encodings, file1)

face_in_camera.py

import cv2
import face_recognition
import ft2
import pickle

def face_in_camera():
    # pickle文件提取
    f1 = open('show_name.pkl', 'rb')
    known_names = pickle.load(f1)
    f = open('show_data.pkl', 'rb', )
    known_face_encodings = pickle.load(f)
    # opencv
    vc = cv2.VideoCapture(0, cv2.CAP_DSHOW)
    while True:
        # 获取视频中每一帧的图片
        ret, img = vc.read()
        if not ret:
            print('没有捕获到视频！')
            break
        locations = face_recognition.face_locations(img)
        face_encodings = face_recognition.face_encodings(img, locations)
        for (top, right, bottom, left), face_encoding in zip(locations, face_encodings):
            matchs = face_recognition.compare_faces(known_face_encodings, face_encoding, tolerance=0.44)
            name = '不在信息表里！'
            for match, known_name in zip(matchs, known_names):
                if match: