基于opencv的人脸识别代码
时间: 2023-10-28 22:06:23 浏览: 49
以下是一个基于OpenCV和Python的人脸识别的示例代码:
```python
import cv2
import os
# 指定人脸检测器
face_detector = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
# 加载训练好的人脸识别模型
face_recognizer = cv2.face.LBPHFaceRecognizer_create()
face_recognizer.read('trainer.yml')
# 指定摄像头编号,如果只有一个摄像头,可以不指定
camera_id = 0
# 打开摄像头
capture = cv2.VideoCapture(camera_id)
# 设置摄像头分辨率
capture.set(cv2.CAP_PROP_FRAME_WIDTH, 640)
capture.set(cv2.CAP_PROP_FRAME_HEIGHT, 480)
# 设置字体
font = cv2.FONT_HERSHEY_SIMPLEX
# 设置人名列表
names = ['unknown', 'person1', 'person2', 'person3']
while True:
# 读取视频帧
ret, frame = capture.read()
# 转换为灰度图像
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
# 检测人脸
faces = face_detector.detectMultiScale(gray, scaleFactor=1.2, minNeighbors=5, minSize=(30, 30))
# 对每个检测到的人脸进行识别
for (x, y, w, h) in faces:
# 绘制人脸矩形框
cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 255, 0), 2)
# 提取人脸图像
face_roi = gray[y:y+h, x:x+w]
# 调整图像大小
face_roi = cv2.resize(face_roi, (200, 200))
# 预测人脸所属的类别
label, confidence = face_recognizer.predict(face_roi)
# 绘制人名标签
name = names[label]
cv2.putText(frame, name, (x, y-20), font, 1, (0, 255, 0), 2)
# 显示视频帧
cv2.imshow('video', frame)
# 按下 q 键退出程序
if cv2.waitKey(1) & 0xFF == ord('q'):
break
# 释放摄像头
capture.release()
# 关闭所有窗口
cv2.destroyAllWindows()
```
需要注意的是,这个示例代码中使用的是LBPH算法进行人脸识别,如果需要改用其他算法,比如Eigenfaces或Fisherfaces,需要更改相应的代码。另外,需要确保训练好的人脸识别模型文件(trainer.yml)和人脸检测器文件(haarcascade_frontalface_default.xml)存在,并且与代码中的路径保持一致。