请用python实现人脸识别
时间: 2023-08-29 10:05:48 浏览: 87
要使用Python实现人脸识别,你需要安装并使用OpenCV和Dlib库。以下是一个基本的人脸识别代码示例:
```python
import cv2
import dlib
# 加载预训练的人脸检测模型
detector = dlib.get_frontal_face_detector()
# 加载预训练的人脸关键点检测模型
predictor = dlib.shape_predictor('shape_predictor_68_face_landmarks.dat')
# 加载人脸识别模型
face_recognition_model = dlib.face_recognition_model_v1('dlib_face_recognition_resnet_model_v1.dat')
# 加载已知人脸的特征向量
known_faces = [
# 第一张人脸的特征向量
[0.1, 0.2, ..., 0.9],
# 第二张人脸的特征向量
[0.2, 0.3, ..., 0.8],
# ...
]
# 打开摄像头
cap = cv2.VideoCapture(0)
while True:
# 读取摄像头捕捉到的图像
ret, frame = cap.read()
# 将图像转为灰度图
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
# 在灰度图上检测人脸
faces = detector(gray)
# 遍历检测到的所有人脸
for face in faces:
# 获取人脸的关键点
landmarks = predictor(gray, face)
# 获取人脸的特征向量
face_descriptor = face_recognition_model.compute_face_descriptor(frame, landmarks)
# 比较人脸的特征向量与已知的人脸特征向量
for known_face in known_faces:
distance = 0
for i in range(len(face_descriptor)):
distance += (face_descriptor[i] - known_face[i]) ** 2
distance = distance ** 0.5
# 如果距离小于一定阈值,则认为是同一个人
if distance < 0.6:
print('Hello, {}!'.format('Tom'))
# 在图像上绘制人脸框和关键点
cv2.rectangle(frame, (face.left(), face.top()), (face.right(), face.bottom()), (0, 255, 0), 2)
for i in range(68):
cv2.circle(frame, (landmarks.part(i).x, landmarks.part(i).y), 2, (0, 0, 255), -1)
# 显示图像
cv2.imshow('frame', frame)
# 按下q键退出循环
if cv2.waitKey(1) & 0xFF == ord('q'):
break
# 释放摄像头并关闭窗口
cap.release()
cv2.destroyAllWindows()
```
请注意,这是一个非常基本的示例,无法处理复杂的场景,例如多人脸检测或遮挡的人脸。在实际应用中,您需要更复杂的算法和更大的数据集来进行人脸识别。
阅读全文