你作为一个机器视觉算法工程师,用python帮我写一个人脸识别,人脸检测的代码
时间: 2023-05-26 07:02:22 浏览: 62
import cv2
import numpy as np
# 加载人脸检测器
faceCascade = cv2.CascadeClassifier("haarcascade_frontalface_default.xml")
# 加载人脸识别模型
recognizer = cv2.face.LBPHFaceRecognizer_create()
recognizer.read("trainer.yml")
# 加载标签
labels = {}
with open("labels.txt", "r") as f:
lines = f.readlines()
for line in lines:
label, name = line.strip().split(":")
labels[int(label)] = name
# 打开摄像头
cap = cv2.VideoCapture(0)
while True:
# 读取一帧图像
ret, frame = cap.read()
# 转换成灰度图像
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
# 检测人脸
faces = faceCascade.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)
# 提取人脸ROI
roi_gray = gray[y:y+h, x:x+w]
roi_color = frame[y:y+h, x:x+w]
# 对人脸ROI进行识别
label, confidence = recognizer.predict(roi_gray)
# 显示识别结果
if confidence < 70:
name = labels[label]
cv2.putText(frame, name, (x, y-20), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
else:
cv2.putText(frame, "Unknown", (x, y-20), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2)
# 显示图像
cv2.imshow("Frame", frame)
# 按下q键退出循环
if cv2.waitKey(1) & 0xFF == ord('q'):
break
# 释放摄像头并关闭窗口
cap.release()
cv2.destroyAllWindows()