人脸识别程序主要通过OpenCV进行实现。具体包括人脸检测和识别两个部分。人脸检测部分采用Haar特征检测算法人脸识别部分采用LBPH算法。具体代码
如下:
人脸检测:
import cv2
# 加载人脸分类器
face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
# 读取图像
img = cv2.imread('test.jpg')
# 转化为灰度图像
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 检测人脸
faces = face_cascade.detectMultiScale(gray, 1.3, 5)
# 绘制矩形框
for (x,y,w,h) in faces:
cv2.rectangle(img,(x,y),(x+w,y+h),(255,0,0),2)
# 显示结果
cv2.imshow('img',img)
cv2.waitKey(0)
cv2.destroyAllWindows()
人脸识别:
import cv2
import os
# 加载人脸分类器
face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
# 加载图像并转化为灰度图像
img = cv2.imread('test.jpg')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 检测人脸
faces = face_cascade.detectMultiScale(gray, 1.3, 5)
# 加载训练好的模型
recognizer = cv2.face.LBPHFaceRecognizer_create()
recognizer.read('train.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
# 遍历每个检测到的人脸
for (x,y,w,h) in faces:
# 提取人脸ROI
roi_gray = gray[y:y+h, x:x+w]
# 进行识别
label, confidence = recognizer.predict(roi_gray)
# 绘制识别结果
name = labels[label]
cv2.rectangle(img, (x,y), (x+w,y+h), (0,255,0), 2)
cv2.putText(img, name, (x,y-10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0,255,0), 2)
# 显示结果
cv2.imshow('img',img)
cv2.waitKey(0)
cv2.destroyAllWindows()
需要注意的是,以上代码仅为示例,具体实现时需要根据实际情况进行调整和优化
原文地址: https://www.cveoy.top/t/topic/fdKJ 著作权归作者所有。请勿转载和采集!