这次用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: