基于 OpenCV 和 Python 的人脸识别签到系统的保姆级教程
创建一个基于 OpenCV 和 Python 的人脸识别签到系统可以是一个有趣且具有挑战性的项目。以下是一个简化的保姆级教程,帮助你入门。
先决条件
在开始之前,请确保你已经安装了以下软件和库:
- Python 3.x
- OpenCV (
opencv-python
) - Numpy
- Pillow(用于图像处理)
- dlib(用于人脸识别)
- face_recognition(基于 dlib 的人脸识别库)
你可以通过以下命令进行安装:
pip install opencv-python numpy pillow dlib face_recognition
步骤 1:设置人脸数据
首先,你需要收集签到人员的脸部图像。可以创建一个名为 known_faces
的文件夹,并在其中存放人员的脸部照片。确保每个图像文件命名为与其对应的姓名相同。
步骤 2:加载并编码已知人脸
创建一个 Python 文件(比如 face_recognition_attendance.py
),并编写以下代码:
import face_recognition
import cv2
import numpy as np
import os
# 加载已知的脸部
known_face_encodings = []
known_face_names = []
known_faces_dir = 'known_faces'
for image_name in os.listdir(known_faces_dir):
image_path = os.path.join(known_faces_dir, image_name)
image = face_recognition.load_image_file(image_path)
face_encoding = face_recognition.face_encodings(image)[0]
known_face_encodings.append(face_encoding)
# 删除文件扩展名
known_face_names.append(os.path.splitext(image_name)[0])
print("已知人脸加载完毕")
步骤 3:实施实时人脸识别
接下来,我们使用电脑摄像头来捕获视频帧并进行人脸识别:
# 打开摄像头
video_capture = cv2.VideoCapture(0)
while True:
# 捕获视频的一帧
ret, frame = video_capture.read()
# 将视频帧从 BGR(OpenCV 默认)转换为 RGB(face_recognition 默认)
rgb_frame = frame[:, :, ::-1]
# 找到当前帧中的所有人脸和脸部编码
face_locations = face_recognition.face_locations(rgb_frame)
face_encodings = face_recognition.face_encodings(rgb_frame, face_locations)
for (top, right, bottom, left), face_encoding in zip(face_locations, face_encodings):
matches = face_recognition.compare_faces(known_face_encodings, face_encoding)
name = "Unknown"
# 如果在人脸数据库中找到匹配的
face_distances = face_recognition.face_distance(known_face_encodings, face_encoding)
best_match_index = np.argmin(face_distances)
if matches[best_match_index]:
name = known_face_names[best_match_index]
# 绘制人脸的矩形框和名字
cv2.rectangle(frame, (left, top), (right, bottom), (0, 0, 255), 2)
cv2.putText(frame, name, (left, top - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 2)
# 显示结果帧
cv2.imshow('Face Recognition', frame)
# 按下 'q' 键退出
if cv2.waitKey(1) & 0xFF == ord('q'):
break
# 释放摄像头和关闭窗口
video_capture.release()
cv2.destroyAllWindows()
步骤 4:实现签到功能
为了实现签到功能,你可以将识别到的人脸信息记录在日志文件中:
import csv
from datetime import datetime
def mark_attendance(name):
with open('attendance_log.csv', 'a', newline='') as file:
writer = csv.writer(file)
now = datetime.now()
dt_string = now.strftime('%Y-%m-%d %H:%M:%S')
writer.writerow([name, dt_string])
# 在识别出名字之后调用 mark_attendance(name)
if matches[best_match_index]:
name = known_face_names[best_match_index]
mark_attendance(name)
注意事项
- 建议在实验室条件下测试脚本,在项目集成之前确保一切功能正常。
- 请注意拍摄环境的光线和背景条件。
- 确保你的
dlib
库正确安装,因为它需要 CMake 和一些编译工具支持。
这样,你就有了一个基本的基于 Python 和 OpenCV 的人脸识别签到系统。可以根据需要进一步拓展,例如增加 UI 界面,或与其他考勤系统集成。