基于vscode人脸识别考勤系统代码
时间: 2023-08-01 09:15:31 浏览: 105
基于人脸识别的考勤系统
以下是一个简单的基于Python语言、OpenCV和face_recognition库的人脸识别考勤系统代码示例,可以在vscode中运行:
```python
import cv2
import face_recognition
import os
import datetime
# 加载已知员工的人脸图像和姓名
known_face_encodings = []
known_face_names = []
for file in os.listdir('known_faces'):
if file.endswith('.jpg') or file.endswith('.jpeg') or file.endswith('.png'):
image = face_recognition.load_image_file(os.path.join('known_faces', file))
face_encoding = face_recognition.face_encodings(image)[0]
known_face_encodings.append(face_encoding)
known_face_names.append(os.path.splitext(file)[0])
# 打开摄像头
video_capture = cv2.VideoCapture(0)
# 初始化一些变量
face_locations = []
face_encodings = []
face_names = []
attendance = {}
while True:
# 读取摄像头中的一帧图像
ret, frame = video_capture.read()
# 缩小图像以加快人脸识别速度
small_frame = cv2.resize(frame, (0, 0), fx=0.25, fy=0.25)
# 将图像从BGR颜色空间转换为RGB颜色空间
rgb_small_frame = small_frame[:, :, ::-1]
# 检测当前帧中的所有人脸
face_locations = face_recognition.face_locations(rgb_small_frame)
face_encodings = face_recognition.face_encodings(rgb_small_frame, face_locations)
face_names = []
for face_encoding in face_encodings:
# 将当前人脸与已知员工的人脸进行比较
matches = face_recognition.compare_faces(known_face_encodings, face_encoding)
name = "Unknown"
# 如果识别出当前人脸属于已知员工,则将其姓名记录在attendance字典中
if True in matches:
first_match_index = matches.index(True)
name = known_face_names[first_match_index]
attendance[name] = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
face_names.append(name)
# 在图像上绘制人脸边框和姓名
for (top, right, bottom, left), name in zip(face_locations, face_names):
# 放大边框以匹配缩小的图像
top *= 4
right *= 4
bottom *= 4
left *= 4
# 在图像上绘制人脸边框
cv2.rectangle(frame, (left, top), (right, bottom), (0, 0, 255), 2)
# 在图像上绘制姓名
cv2.rectangle(frame, (left, bottom - 35), (right, bottom), (0, 0, 255), cv2.FILLED)
font = cv2.FONT_HERSHEY_DUPLEX
cv2.putText(frame, name, (left + 6, bottom - 6), font, 1.0, (255, 255, 255), 1)
# 在窗口中显示图像
cv2.imshow('Video', frame)
# 按下'q'键退出程序
if cv2.waitKey(1) & 0xFF == ord('q'):
break
# 释放摄像头并关闭窗口
video_capture.release()
cv2.destroyAllWindows()
# 输出考勤记录
print(attendance)
```
这个示例代码的工作原理如下:
1. 加载已知员工的人脸图像和姓名,并对其进行编码。
2. 打开摄像头并读取其输出的图像帧。
3. 对当前帧中的所有人脸进行检测和编码。
4. 将当前人脸与已知员工的人脸进行比较,如果识别出当前人脸属于已知员工,则将其姓名记录在attendance字典中。
5. 在图像上绘制人脸边框和姓名,并在窗口中显示图像。
6. 按下'q'键退出程序,释放摄像头并关闭窗口。
7. 输出考勤记录。
希望这个示例代码能够帮助您构建您的人脸识别考勤系统。请注意,这只是一个简单的示例,您需要根据您的实际需求进行修改和扩展。
阅读全文