
本文旨在解决基于 OpenCV 和 face_recognition 库构建的人脸识别考勤系统中,重复记录考勤信息的问题。通过分析代码逻辑和文件读写操作,提供两种优化方案,确保考勤记录的准确性和效率。针对初学者,本文提供详细的代码示例和解释,帮助读者理解并解决实际问题。
在人脸识别考勤系统中,一个常见的问题是当摄像头持续检测到同一张脸时,系统会重复将该人员的考勤信息写入 CSV 文件。这不仅导致数据冗余,也影响了考勤记录的准确性。本文将探讨如何解决这个问题,并提供两种优化方案,提升考勤系统的稳定性和效率。
原始代码的问题在于 markAttendance 函数中的逻辑。该函数每次被调用时,都会打开 Attendance.csv 文件,读取所有已记录的姓名,然后检查当前检测到的姓名是否已存在。如果不存在,则将该姓名和当前时间写入文件。关键问题在于,这个检查和写入操作是在主循环中进行的,这意味着只要摄像头检测到人脸,markAttendance 函数就会被调用,从而导致重复写入。
第一个解决方案是将判断姓名是否已存在的逻辑移到循环外部。具体来说,首先在函数内部读取所有已存在的姓名,存储到一个列表中。然后在循环结束后,再判断当前姓名是否在列表中。如果不在,则写入文件。
def markAttendance(name):
with open('Attendance.csv', 'r+') as f:
nameList = []
for line in f:
entry = line.split(',')
nameList.append(entry[0])
if name not in nameList:
dt = datetime.now().strftime('%H:%M:%S')
f.writelines(f'\n{name},{dt}')代码解释:
注意事项:
第二个解决方案是在内存中缓存已记录的姓名列表。这意味着在程序启动时,读取 CSV 文件中的所有姓名,存储到一个列表中。然后,在每次检测到人脸时,直接在内存中查找该姓名是否已存在。如果不存在,则将该姓名添加到内存列表,并写入文件。
def readNames():
with open('Attendance.csv', 'r') as f:
nameList = []
for line in f:
entry = line.split(',')
nameList.append(entry[0])
return nameList
def markAttendance(name, nameList):
if name not in nameList:
nameList.append(name)
with open('Attendance.csv', 'a') as f:
dt = datetime.now().strftime('%H:%M:%S')
f.writelines(f'\n{name},{dt}')
# 初始化姓名列表
nameList = readNames()
while True:
# ... (人脸识别代码) ...
if matches[matchIndex]:
# ...
markAttendance(name, nameList)代码解释:
代码修改示例(将上述代码整合到原代码中):
import os
import cv2
import numpy as np
import face_recognition
from datetime import datetime
path = 'MainImages'
images = []
classNames = []
myList = os.listdir(path)
for cl in myList:
curImg = cv2.imread(f'{path}/{cl}')
images.append(curImg)
classNames.append(os.path.splitext(cl)[0])
def findEncodings(images):
encodeList = []
for img in images:
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
encode = face_recognition.face_encodings(img)[0]
encodeList.append(encode)
return encodeList
def readNames():
with open('Attendance.csv', 'r') as f:
nameList = []
for line in f:
entry = line.split(',')
nameList.append(entry[0])
return nameList
def markAttendance(name, nameList):
if name not in nameList:
nameList.append(name)
with open('Attendance.csv', 'a') as f:
dt = datetime.now().strftime('%H:%M:%S')
f.writelines(f'\n{name},{dt}')
encodeListKnown = findEncodings(images)
print('Encoding Complete')
cap = cv2.VideoCapture(0)
nameList = readNames() # 初始化姓名列表
while True:
success, img = cap.read()
imgS = cv2.resize(img, (0, 0), None, 0.25, 0.25)
imgS = cv2.cvtColor(imgS, cv2.COLOR_BGR2RGB)
facesCurFrame = face_recognition.face_locations(imgS)
encodesCurFrame = face_recognition.face_encodings(imgS, facesCurFrame)
for encodeFace, faceLoc in zip(encodesCurFrame, facesCurFrame):
matches = face_recognition.compare_faces(encodeListKnown, encodeFace)
faceDis = face_recognition.face_distance(encodeListKnown, encodeFace)
matchIndex = np.argmin(faceDis)
if matches[matchIndex]:
name = classNames[matchIndex].upper()
y1, x2, y2, x1 = faceLoc
y1, x2, y2, x1 = y1*4, x2*4, y2*4, x1*4
cv2.rectangle(img, (x1, y1), (x2, y2), (0, 255, 0), 2)
cv2.rectangle(img, (x1, y2-35), (x2, y2), (0, 255, 0), cv2.FILLED)
cv2.putText(img, name, (x1+6, y2-6), cv2.FONT_HERSHEY_COMPLEX, 1, (255, 255, 255), 2)
markAttendance(name, nameList) # 传入姓名列表
cv2.imshow('Webcam', img)
cv2.waitKey(1)注意事项:
本文提供了两种解决人脸识别考勤系统重复记录问题的方案。第一种方案通过将判断逻辑移到循环外部,避免了重复写入。第二种方案通过在内存中缓存姓名列表,提高了效率。选择哪种方案取决于实际应用场景和数据量大小。对于数据量较小的应用,第一种方案可能更简单易懂。对于数据量较大的应用,第二种方案可能更高效。在实际应用中,可以根据具体情况选择合适的方案,并进行适当的优化。
以上就是解决人脸识别考勤系统重复记录问题:一份详细教程的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号