赞
踩
这个人脸识别考勤签到系统是基于大佬的人脸识别陌生人报警系统二次开发的。
项目使用Python实现,基于OpenCV框架进行人脸识别和摄像头硬件调用,同时也用OpenCV工具包处理图片。交互界面使用pyqt5实现。
该系统实现了从学生信息输入、人脸数据录入、人脸数据训练,学生信息多条件搜索、修改,多选删除,人脸数据训练,人脸识别、追踪、签到等完整流程的各项功能。甚至允许生成签到表格和导出Excel格式签到表。
根据功能分配,系统分为三个部分实现各部分流程,
源码下载地址https://download.csdn.net/download/u013749113/87897481
本系统在原始项目的框架基础上做了大量修改,针对系统功能的不同以及部分模块实现的不完整做了补充和优化。原项目在功能上实现了完整的人脸数据录入过程,并已经存在数据管理、数据录入、核心(Core)三个基本模块。
(1)人脸识别实现
在主要部分【核心】的实现上,原项目采用LBPH实现了人脸识别的基本功能,并使用haar-like实现人脸位置捕获,使用dlib目标追踪器实现同一画面下多个人脸的目标追踪,尝试主要追踪,次要捕获的方式优化人脸捕获的过程。识别速度与画面帧率很高,
但识别准确率并不理想,并且严重受到光照条件影响。【实际使用时证明,若录入人脸数据时存在单方面光源照射,在识别时,光源位置一旦改变就完全识别不出来了】
并且录入数据时需要大量人脸数据集【100张人脸图以上】,才能获得较高的识别置信度。
二次开发后,保留原本速度较快的LBPH人脸识别,新增效果更好的dlib_face_recognition_resnet_model深度学习残差网络识别模型。实际使用时发现,虽然实时视频流帧率明显降低,但是识别准确率大大提高。
(2)摄像头启动与关闭
在原项目中,在摄像头调用的设计上,录入端和监控端同样都是用了摄像头,并进行人脸捕获,但监控端只允许打开和关闭一次摄像头,之后就禁用了摄像头控制按钮,而录入端则允许随意多次的打开和关闭摄像头。
其原因是具体实现上,因为监控端需要启动单独线程执行人脸识别任务,而录入端没有人脸识别的复杂处理过程,在一个线程中就可以实现需要的功能。监控端开启摄像头后,同时启动人脸识别线程处理单帧图像,并启动报警监听线程,设计思路上是报警线程一旦开启,全程保持监听,直到程序关闭,而关闭摄像头时代码实现上只关闭,也只能够关闭人脸识别线程,对于报警监听线程无法控制。导致如果容许摄像头二次开启,人脸识别线程能够启动,但作为同一个线程实例的报警监听模块将被二次启动,导致程序出错。
这里的人脸识别线程继承自QTread类,允许使用.stop()函数控制线程实例的开始与终止,而报警监听模块则是普通的threading.Thread(),开启后无法控制关闭,导致出现上述问题。改进方法是将报警监听线程使用QTread实现,使用与人脸识别线程相同的控制方法,摄像头开启时启动,摄像头关闭时终止,实现二次启动功能。
(3)报警系统改为签到系统
因为系统的功能修改前后有所改变,但是实现的技术实际上是一样的。原始项目的报警功能实际上可以作为签到系统的监听线程。原始代码逻辑为,发现置信度低于阈值的脸即陌生人脸,随机人脸识别线程通过通信队列告知报警系统执行截图拍摄、消息推送、报警响铃等功能。改变为签到系统后,人脸识别线程将置信度阈值以上【即数据库中存在并认为可信的脸】,且匹配度最高的人脸作为签到信号,通过队列通信发送给签到线程,并执行之前没有的数据库录入操作。
简单来说就是,之前是把认不出来的脸进行记录和报警,转变为认出来且最像的人脸进行记录和执行声音提示。
(4)签到表格的创建
这个属于签到系统特有的新增功能,说道考勤签到就一定会想到教师用来点名的签到表,因此该签到系统需要生成一个包含当前课程需要签到的所有学生信息,通过预处理进行创建,并在人脸识别过程中实时修改记录签到信息,主要内容为学生姓名、学号、签到时间。该功能内建了一个基于MySQL查询的QTableWidget控件,用于方便用户之间从数据库中选择并创建需要签到的学生名单。
(1)学生信息管理
原始的管理端非常简陋,存储信息仅包含学生姓名、faceID、学号这样的普通信息,想要查询学生信息也只能进行单人查询,并且只能通过学号,查询的唯一目的就是通过学号删除查询结果。因此无论是查询还是删除都非常简陋。甚至根本不存在修改信息的功能。
修改后的管理端新增了多条件模糊查询,并且大大增加了信息维数,允许直接双击修改学生信息,并实时同步到数据库中。同时删除信息的功能也和信息查询分离开来,信息查询结果动态显示在QTableWidget控件中【与原项目使用的技术相同,只是做了更大的功能扩展】。支持用户多选删除,不不必“查一个,删一个”。
(2)人脸数据训练
原项目中的人脸数据训练功能同样集成在管理端中,原作者自定义了一个读取数据集并将其与学生通过faceID进行唯一匹配的函数,也就是LBPH数据训练中的人脸数据(faces)与分类标签(labels)。读取并整理成LBPH.train()所需要的数据结构后即可直接将数据作为参数调用封装好的函数进行训练。
值得一提的是,LBPH的数据训练非常快,即使在三百人的,每人人脸数据集平均20张的情况下,训练时间依然能控制在十几秒内【但是结果其实并不好】,因此原作者并没有将其作为单独的线程来执行,而是直接甩给用户一个提示:训练期间系统窗口可能会无响应,请耐心等待。。。
这样的结果是导致用户体验极差,会以为程序崩了,其实只是训练计算时间过长,导致windows消息监听一段时间内无回应,被认为是程序无响应。。。
二次开发后我将其独立为一个线程单独执行,并为执行函数增加了进度条,让用户直观的看到训练过程。顺便,在代码实现过程中,发现大量时间实际上消耗在图片数据集的读取上,训练的过程反倒没有那么久,因此进度条实际展示的是读取过程,逻辑上是先进行的数据读取,再计算特征值,然后继续读取下一个数据集,因此将数据读取作为进度来衡量不会有时间上的偏差。
这里就简单贴一下数据录入端的代码。项目完整代码还请移步本文开始位置的链接。
#!/usr/bin/env python3 # Author: kuronekonano <god772525182@gmail.com> # 人脸信息录入 import re import string import time import cv2 import pymysql import shutil from PyQt5.QtCore import QTimer, QRegExp, pyqtSignal, QThread from PyQt5.QtGui import QImage, QPixmap, QIcon, QRegExpValidator, QTextCursor from PyQt5.QtWidgets import QDialog, QApplication, QWidget, QMessageBox, QFileDialog, QProgressBar from PyQt5.uic import loadUi import logging import logging.config import queue import threading import os import sys import xlrd import random from datetime import datetime # 用户取消了更新数据库操作 class OperationCancel(Exception): pass # 采集过程中出现干扰 class RecordDisturbance(Exception): pass class DataRecordUI(QWidget): receiveLogSignal = pyqtSignal(str) messagebox_signal = pyqtSignal(dict) # 日志队列 logQueue = queue.Queue() def __init__(self): super(DataRecordUI, self).__init__() loadUi('./ui/DataRecord.ui', self) # 读取UI布局 self.setWindowIcon(QIcon('./icons/icon.png')) self.setFixedSize(1528, 856) # OpenCV # 摄像头 self.cap = cv2.VideoCapture() # 分类器 self.faceCascade = cv2.CascadeClassifier('./haarcascades/haarcascade_frontalface_default.xml') # 图像捕获 self.isExternalCameraUsed = False self.useExternalCameraCheckBox.stateChanged.connect( lambda: self.useExternalCamera(self.useExternalCameraCheckBox)) self.startWebcamButton.toggled.connect(self.startWebcam) self.startWebcamButton.setCheckable(True) # 定时器 self.timer = QTimer(self) self.timer.timeout.connect(self.updateFrame) # 人脸检测 self.isFaceDetectEnabled = False self.enableFaceDetectButton.toggled.connect(self.enableFaceDetect) self.enableFaceDetectButton.setCheckable(True) # 数据库 # self.database = 'users' self.datasets = './datasets' self.isDbReady = False self.initDbButton.setIcon(QIcon('./icons/warning.png')) self.initDbButton.clicked.connect(self.initDb) # 用户信息 self.isUserInfoReady = False self.userInfo = {'stu_id': '', 'cn_name': '', 'en_name': '', 'stu_grade': '', 'stu_class': '', 'stu_sex': '', 'major': ''} self.addOrUpdateUserInfoButton.clicked.connect(self.addOrUpdateUserInfo) self.migrateToDbButton.clicked.connect(self.migrateToDb) # 插入新数据按键绑定 # 人脸采集 self.startFaceRecordButton.clicked.connect( lambda: self.startFaceRecord(self.startFaceRecordButton)) # 开始人脸采集按钮绑定,并传入按钮本身用于结束状态控制 # self.startFaceRecordButton.setCheckable(True) self.faceRecordCount = 0 # 已采集照片计数器 self.minFaceRecordCount = 100 # 最少采集照片数量 self.isFaceDataReady = False self.isFaceRecordEnabled = False self.enableFaceRecordButton.clicked.connect(self.enableFaceRecord) # 按键绑定录入单帧图像 # 日志系统 self.receiveLogSignal.connect(lambda log: self.logOutput(log)) # pyqtsignal信号绑定 self.messagebox_signal.connect(lambda log: self.message_output(log)) self.logOutputThread = threading.Thread(target=self.receiveLog, daemon=True) self.logOutputThread.start() # 批量导入 self.isImage_path_ready = False # self.ImagepathButton.clicked.connect(self.import_images_data) # 使用同一线程会导致窗口无响应 self.ImagepathButton.clicked.connect(self.import_image_thread) # 使用多线程实现图片导入 self.isExcel_path_ready = False self.ExcelpathButton.clicked.connect(self.import_excel_data) self.ImportPersonButton.clicked.connect(self.person_import_thread) @staticmethod def connect_to_sql(): conn = pymysql.connect(host='localhost', user='root', password='******', db='mytest', port=3306, charset='utf8') cursor = conn.cursor() return conn, cursor # 单人导入图片集【主线程】弃用 def import_person_imageset(self): if self.isUserInfoReady: # 学生信息确认 stu_id = self.userInfo.get('stu_id') self.ImportPersonButton.setIcon(QIcon('./icons/success.png')) image_paths = QFileDialog.getOpenFileNames(self, '选择图片', "./", 'JEPG files(*.jpg);;PNG files(*.PNG)') if not os.path.exists('{}/stu_{}'.format(self.datasets, stu_id)): os.makedirs('{}/stu_{}'.format(self.datasets, stu_id)) image_paths = image_paths[0] for index, path in enumerate(image_paths): try: img = cv2.imread(path) gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) # 灰度图 faces = self.faceCascade.detectMultiScale(gray, 1.3, 5, minSize=(90, 90)) # 分类器侦测人脸 if len(faces) == 0: self.logQueue.put('图片{}中没有检测到人脸!'.format(path)) continue for (x, y, w, h) in faces: if len(faces) > 1: raise RecordDisturbance cv2.imwrite('{}/stu_{}/img.{}-{}.jpg'.format(self.datasets, stu_id, index, ''.join( random.sample(string.ascii_letters + string.digits, 4))), img[y - 20:y + h + 20, x - 20:x + w + 20]) # 灰度图的人脸区域 except RecordDisturbance: logging.error('检测到多张人脸或环境干扰') self.logQueue.put('Warning:检测到图片{}存在多张人脸或环境干扰,已忽略。'.format(path)) continue except Exception as e: logging.error('写入人脸图像文件到计算机过程中发生异常') self.logQueue.put('Error:无法保存人脸图像,导入该图片失败') print(e) self.migrateToDbButton.setEnabled(True) # 允许提交至数据库 self.isFaceDataReady = True else: self.ImportPersonButton.setIcon(QIcon('./icons/error.png')) self.ImportPersonButton.setChecked(False) self.logQueue.put('Error:操作失败,系统未检测到有效的用户信息') # 表格导入学生信息 def import_excel_data(self): excel_paths = QFileDialog.getOpenFileNames(self, '选择表格', "./", 'EXCEL 文件 (*.xlsx;*.xls;*.xlm;*.xlt;*.xlsm;*.xla)') excel_paths = excel_paths[0] conn, cursor = self.connect_to_sql() error_count = 0 for path in excel_paths: sheets_file = xlrd.open_workbook(path) for index, sheet in enumerate(sheets_file.sheets()): self.logQueue.put("正在读取文件:" + str(path) + "的第" + str(index) + "个sheet表的内容...") for row in range(sheet.nrows): row_data = sheet.row_values(row) if row_data[1] == '姓名': continue self.userInfo['stu_id'] = row_data[4] self.userInfo['cn_name'] = row_data[1] self.userInfo['en_name'] = row_data[0] self.userInfo['stu_grade'] = '20' + self.userInfo['stu_id'][:2] self.userInfo['stu_class'] = row_data[3].rsplit('-', 1)[1] self.userInfo['stu_sex'] = row_data[5] self.userInfo['major'] = row_data[2] self.userInfo['province'] = row_data[-1] self.userInfo['nation'] = row_data[-2] # print(self.userInfo) try: stu_id = row_data[4] if not os.path.exists('{}/stu_{}'.format(self.datasets, stu_id)): os.makedirs('{}/stu_{}'.format(self.datasets, stu_id)) db_user_count = self.commit_to_database(cursor) self.dbUserCountLcdNum.display(db_user_count) # 数据库人数计数器 except OperationCancel: pass except Exception as e: print(e) logging.error('读写数据库异常,无法向数据库插入/更新记录') self.logQueue.put('Error:读写数据库异常,同步失败') error_count += 1 self.logQueue.put('导入完毕!其中导入失败 {} 条信息'.format(error_count)) cursor.close() conn.commit() conn.close() # 启用新线程导入图片,并添加进度条 def import_image_thread(self): self.image_paths = QFileDialog.getOpenFileNames(self, '选择图片', "./", 'JEPG files(*.jpg);;PNG files(*.PNG)') self.image_paths = self.image_paths[0] if len(self.image_paths) != 0: # 点击导入但是没有选择文件时不需启动线程 progress_bar = ActionsImportImage(self) print('import success!') # 启用新线程 单人图片导入 使用进度条 def person_import_thread(self): if self.isUserInfoReady: # 学生信息确认 stu_id = self.userInfo.get('stu_id') self.ImportPersonButton.setIcon(QIcon('./icons/success.png')) image_paths = QFileDialog.getOpenFileNames(self, '选择图片', "./", 'JEPG files(*.jpg);;PNG files(*.PNG)') self.image_paths = image_paths[0] if len(self.image_paths) != 0: # 点击导入但是没有选择文件时不需启动线程 if not os.path.exists('{}/stu_{}'.format(self.datasets, stu_id)): os.makedirs('{}/stu_{}'.format(self.datasets, stu_id)) progress_bar = ActionsPersonImport(self) self.migrateToDbButton.setEnabled(True) # 允许提交至数据库 self.isFaceDataReady = True else: self.ImportPersonButton.setIcon(QIcon('./icons/error.png')) self.ImportPersonButton.setChecked(False) self.logQueue.put('Error:操作失败,系统未检测到有效的用户信息') # 图片批量导入【主线程】弃用 def import_images_data(self): image_paths = QFileDialog.getOpenFileNames(self, '选择图片', "./", 'JEPG files(*.jpg);;PNG files(*.PNG)') image_paths = image_paths[0] error_count = 0 self.logQueue.put('开始读取图片数据...') for index, path in enumerate(image_paths): stu_id = os.path.split(path)[1].split('.')[0] # print(stu_id) if not os.path.exists('{}/stu_{}'.format(self.datasets, stu_id)): text = '命名错误!' informativeText = '<b>文件 <font color=red>{}</font> 存在问题,数据库中没有以该图片名为学号的用户。</b>'.format(path) DataRecordUI.callDialog(QMessageBox.Critical, text, informativeText, QMessageBox.Ok) error_count += 1 continue dstpath = '{}/stu_{}/img.{}.jpg'.format(self.datasets, stu_id, stu_id + '-0') try: shutil.copy(path, dstpath) except: text = '命名格式错误!' informativeText = '<b>文件 <font color=red>{}</font> 命名格式不正确。</b>'.format(path) DataRecordUI.callDialog(QMessageBox.Critical, text, informativeText, QMessageBox.Ok) error_count += 1 self.logQueue.put('图片批量导入完成!其中导入失败 {} 张图片'.format(error_count)) # 是否使用外接摄像头 def useExternalCamera(self, useExternalCameraCheckBox): if useExternalCameraCheckBox.isChecked(): self.isExternalCameraUsed = True else: self.isExternalCameraUsed = False # 打开/关闭摄像头 def startWebcam(self, status): if status: if not self.cap.isOpened(): camID = 1 if self.isExternalCameraUsed else 0 + cv2.CAP_DSHOW self.cap.open(camID) self.cap.set(cv2.CAP_PROP_FRAME_WIDTH, 640) self.cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 480) ret, frame = self.cap.read() # 获取摄像头调用结果 if not ret: logging.error('无法调用电脑摄像头{}'.format(camID)) self.logQueue.put('Error:初始化摄像头失败') self.cap.release() self.startWebcamButton.setIcon(QIcon('./icons/error.png')) self.startWebcamButton.setChecked(False) else: self.timer.start(5) self.enableFaceDetectButton.setEnabled(True) self.startWebcamButton.setIcon(QIcon('./icons/success.png')) self.startWebcamButton.setText('关闭摄像头') else: if self.cap.isOpened(): if self.timer.isActive(): self.timer.stop() self.cap.release() self.faceDetectCaptureLabel.clear() self.faceDetectCaptureLabel.setText('<font color=red>摄像头未开启</font>') self.startWebcamButton.setText('打开摄像头') self.enableFaceDetectButton.setEnabled(False) self.startWebcamButton.setIcon(QIcon()) # 开启/关闭人脸检测 def enableFaceDetect(self, status): if self.cap.isOpened(): if status: self.enableFaceDetectButton.setText('关闭人脸检测') self.isFaceDetectEnabled = True else: self.enableFaceDetectButton.setText('开启人脸检测') self.isFaceDetectEnabled = False # 采集当前捕获帧 def enableFaceRecord(self): if not self.isFaceRecordEnabled: self.isFaceRecordEnabled = True # 开始/结束采集人脸数据 def startFaceRecord(self, startFaceRecordButton): if startFaceRecordButton.text() == '开始采集人脸数据': # 只能用==判断,不能用is if self.isFaceDetectEnabled: if self.isUserInfoReady: # 学生信息确认 self.addOrUpdateUserInfoButton.setEnabled(False) # 采集人脸数据时禁用修改学生信息 if not self.enableFaceRecordButton.isEnabled(): # 启用单帧采集按钮 self.enableFaceRecordButton.setEnabled(True) self.enableFaceRecordButton.setIcon(QIcon()) self.startFaceRecordButton.setIcon(QIcon('./icons/success.png')) self.startFaceRecordButton.setText('结束当前人脸采集') # 开始采集按钮状态修改为结束采集 else: self.startFaceRecordButton.setIcon(QIcon('./icons/error.png')) self.startFaceRecordButton.setChecked(False) self.logQueue.put('Error:操作失败,系统未检测到有效的用户信息') else: self.startFaceRecordButton.setIcon(QIcon('./icons/error.png')) self.logQueue.put('Error:操作失败,请开启人脸检测') else: # 根据按钮文本信息判断是结束采集还是开始采集 if self.faceRecordCount < self.minFaceRecordCount: text = '系统当前采集了 <font color=blue>{}</font> 帧图像,采集数据过少会导致较大的识别误差。'.format(self.faceRecordCount) informativeText = '<b>请至少采集 <font color=red>{}</font> 帧图像。</b>'.format(self.minFaceRecordCount) DataRecordUI.callDialog(QMessageBox.Information, text, informativeText, QMessageBox.Ok) else: text = '系统当前采集了 <font color=blue>{}</font> 帧图像,继续采集可以提高识别准确率。'.format(self.faceRecordCount) informativeText = '<b>你确定结束当前人脸采集吗?</b>' ret = DataRecordUI.callDialog(QMessageBox.Question, text, informativeText, QMessageBox.Yes | QMessageBox.No, QMessageBox.No) if ret == QMessageBox.Yes: self.isFaceDataReady = True # 结束采集,人脸数据准备完毕 if self.isFaceRecordEnabled: self.isFaceRecordEnabled = False self.enableFaceRecordButton.setEnabled(False) # 结束采集,单帧采集按钮禁用 self.enableFaceRecordButton.setIcon(QIcon()) self.startFaceRecordButton.setText('开始采集人脸数据') # 修改按钮文本为开始状态 self.startFaceRecordButton.setEnabled(False) # 不可重新开始采集 self.startFaceRecordButton.setIcon(QIcon()) self.migrateToDbButton.setEnabled(True) # 允许提交至数据库 # 定时器,实时更新画面 def updateFrame(self): ret, frame = self.cap.read() # frame = cv2.flip(frame, 1) # 水平翻转图片 if ret: # self.displayImage(frame) # ?两次输出? if self.isFaceDetectEnabled: # 人脸检测 detected_frame = self.detectFace(frame) self.displayImage(detected_frame) else: self.displayImage(frame) # 检测人脸 def detectFace(self, frame): gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) # 灰度图 faces = self.faceCascade.detectMultiScale(gray, 1.3, 5, minSize=(90, 90)) # 分类器侦测人脸 # 1.image为输入的灰度图像 # 2.objects为得到被检测物体的矩形框向量组 # 3.scaleFactor为每一个图像尺度中的尺度参数,默认值为1.1。scale_factor参数可以决定两个不同大小的窗口扫描之间有多大的跳跃, # 这个参数设置的大,则意味着计算会变快,但如果窗口错过了某个大小的人脸,则可能丢失物体。 # 4.minNeighbors参数为每一个级联矩形应该保留的邻近个数,默认为3。 # minNeighbors控制着误检测,默认值为3表明至少有3次重叠检测,我们才认为人脸确实存。 # 6.cvSize()指示寻找人脸的最小区域。设置这个参数过大,会以丢失小物体为代价减少计算量。 stu_id = self.userInfo.get('stu_id') # 遍历所有人脸,只允许有一个人的脸 for (x, y, w, h) in faces: if self.isFaceRecordEnabled: try: # 创建学号对应的图片数据集 if not os.path.exists('{}/stu_{}'.format(self.datasets, stu_id)): os.makedirs('{}/stu_{}'.format(self.datasets, stu_id)) if len(faces) > 1: raise RecordDisturbance cv2.imwrite('{}/stu_{}/img.{}.jpg'.format(self.datasets, stu_id, self.faceRecordCount + 1), frame[y - 20:y + h + 20, x - 20:x + w + 20]) # 灰度图的人脸区域 except RecordDisturbance: self.isFaceRecordEnabled = False logging.error('检测到多张人脸或环境干扰') self.logQueue.put('Warning:检测到多张人脸或环境干扰,请解决问题后继续') self.enableFaceRecordButton.setIcon(QIcon('./icons/warning.png')) continue except Exception as e: logging.error('写入人脸图像文件到计算机过程中发生异常') self.enableFaceRecordButton.setIcon(QIcon('./icons/error.png')) self.logQueue.put('Error:无法保存人脸图像,采集当前捕获帧失败') else: self.enableFaceRecordButton.setIcon(QIcon('./icons/success.png')) self.faceRecordCount = self.faceRecordCount + 1 self.isFaceRecordEnabled = False # 单帧拍摄完成后马上关闭 self.faceRecordCountLcdNum.display(self.faceRecordCount) # 更新采集数量 cv2.rectangle(frame, (x - 5, y - 10), (x + w + 5, y + h + 10), (0, 0, 255), 2) # 红色追踪框 return frame
源码下载地址https://download.csdn.net/download/u013749113/87897481
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。