Python PyQt5 人脸识别软件

# !/usr/bin/python3# -*- coding:utf-8 -*-"""@author: JHC000abc@gmail.com@file: untitled_ctrl.py@time: 2023/8/6 16:29@desc:"""import copyimport threadingimport timeimport tracebackimport uuidimport osimport cv2from gui.ui.untitled import Ui_widgetfrom PyQt5 import QtWidgetsfrom PyQt5.QtCore import pyqtSignalfrom PyQt5.QtGui import QPixmap, QImagefrom threading import Threadfrom queue import Queueimport numpyimport PyQt5class UntitledCtrl(QtWidgets.QWidget):""""""single_frame = pyqtSignal(PyQt5.QtGui.QImage)def __init__(self, rootpath):super(UntitledCtrl, self).__init__()self.rootpath = rootpathself.resize(980, 521)self.ui = Ui_widget()self.ui.setupUi(self)print(rootpath)self.img_que = Queue()self.ui.pushButton_cut.hide()self._flag()self._slot()self._thread()def _thread(self):t = Thread(target=self.camera)t.start()t2 = Thread(target=self.show_img)t2.start()def _flag(self):self.flag_start = Trueself.flag_face = Trueself.cap = Noneself.pause = Falseself.frame = ""self.cut = ""def _slot(self):self.ui.pushButton_start.clicked.connect(self.start_pause)self.ui.pushButton_check_face.clicked.connect(self.face_check)self.single_frame.connect(self.th_show)def th_show(self, img):if img:try:self.ui.label_camera.setPixmap(QPixmap(img))except BaseException:passdef show_img(self):""":return:"""while True:try:if not self.img_que.empty():img, nums = self.img_que.get()for i in range(nums):self.single_frame.emit(img)time.sleep(0.01)except Exception as e:print("e2", e)time.sleep(0.01)def add_face_box(self):""":return:"""flag = Falseif self.flag_start:if self.frame is not None:self.cut = copy.deepcopy(self.frame)try:if self.flag_face:for x, y, w, h in self.get_face_opt():try:self.cut = cv2.rectangle(img=self.cut,pt1=(x, y),pt2=(x + w, y + h),color=(0, 0, 255),thickness=2,lineType=cv2.INTER_LINEAR,)flag = Trueexcept BaseException:passimage = cv2.cvtColor(self.cut, cv2.COLOR_RGB2BGR)self.cut = QImage(image.data,image.shape[1],image.shape[0],QImage.Format_RGB888)if flag:self.img_que.put([self.cut, 30])else:self.img_que.put([self.cut, 1])except Exception as e:print("e1", str(traceback.print_exc()))def camera(self):""":return:"""try:if not self.cap:self.cap = cv2.VideoCapture(0)# 使用默认摄像头if not self.cap.isOpened():print("无法打开摄像头")exit()while not self.pause:ret, self.frame = self.cap.read()if not ret:print("无法读取视频流")breakelse:self.add_face_box()self.cap.release()except Exception as e:print(traceback.print_exc())def get_face_opt(self):""":return:"""face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades +'haarcascade_frontalface_default.xml')try:faces = face_cascade.detectMultiScale(self.cut, 1.1, 4)for (x, y, w, h) in faces:yield x, y, w, hexcept BaseException:passdef start_pause(self):""":return:"""if self.flag_start:self.ui.pushButton_start.setText("暂停")self.flag_start = Falseelse:self.ui.pushButton_start.setText("开始")self.flag_start = Truedef face_check(self):""":return:"""if self.flag_face:self.flag_face = Falseself.ui.pushButton_check_face.setText("暂停识别")else:self.flag_face = Trueself.ui.pushButton_check_face.setText("识别人脸")def save(self):file = "gui/res/{}.png".format(uuid.uuid4())cv2.imencode(".{}".format(".{}".format(os.path.splitext(file)[-1])), self.frame)[1].tofile(file)

# -*- coding: utf-8 -*-# Form implementation generated from reading ui file 'untitled.ui'## Created by: PyQt5 UI code generator 5.15.9## WARNING: Any manual changes made to this file will be lost when pyuic5 is# run again.Do not edit this file unless you know what you are doing.from PyQt5 import QtCore, QtGui, QtWidgetsclass Ui_widget(object):def setupUi(self, widget):widget.setObjectName("widget")# widget.resize(836, 521)self.horizontalLayout = QtWidgets.QHBoxLayout(widget)self.horizontalLayout.setObjectName("horizontalLayout")self.stackedWidget = QtWidgets.QStackedWidget(widget)self.stackedWidget.setObjectName("stackedWidget")self.page = QtWidgets.QWidget()self.page.setObjectName("page")self.gridLayout = QtWidgets.QGridLayout(self.page)self.gridLayout.setObjectName("gridLayout")self.horizontalLayout_2 = QtWidgets.QHBoxLayout()self.horizontalLayout_2.setObjectName("horizontalLayout_2")self.label_camera = QtWidgets.QLabel(self.page)self.label_camera.setStyleSheet("border-width: 3px;\n""border-style: dot-dot-dash;\n""border-color: purple;\n""border-radius: 50px;\n""")self.label_camera.setText("")self.label_camera.setObjectName("label_camera")self.horizontalLayout_2.addWidget(self.label_camera)spacerItem = QtWidgets.QSpacerItem(40, 20, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum)self.horizontalLayout_2.addItem(spacerItem)self.frame = QtWidgets.QFrame(self.page)self.frame.setStyleSheet("border-width: 3px;\n""border-style: dot-dot-dash;\n""border-color: purple;\n""border-radius: 50px;\n""")self.frame.setFrameShape(QtWidgets.QFrame.StyledPanel)self.frame.setFrameShadow(QtWidgets.QFrame.Raised)self.frame.setObjectName("frame")self.pushButton_start = QtWidgets.QPushButton(self.frame)self.pushButton_start.setGeometry(QtCore.QRect(100, 20, 93, 28))self.pushButton_start.setStyleSheet("border-style: solid;\n""border-color:purple;")self.pushButton_start.setObjectName("pushButton_start")self.pushButton_cut = QtWidgets.QPushButton(self.frame)self.pushButton_cut.setGeometry(QtCore.QRect(100, 80, 93, 28))self.pushButton_cut.setStyleSheet("border-style: solid;\n""border-color:purple;")self.pushButton_cut.setObjectName("pushButton_cut")self.pushButton_check_face = QtWidgets.QPushButton(self.frame)self.pushButton_check_face.setGeometry(QtCore.QRect(100, 140, 93, 28))self.pushButton_check_face.setStyleSheet("border-style: solid;\n""border-color:purple;")self.pushButton_check_face.setObjectName("pushButton_check_face")self.horizontalLayout_2.addWidget(self.frame)self.horizontalLayout_2.setStretch(0, 150)self.horizontalLayout_2.setStretch(1, 1)self.horizontalLayout_2.setStretch(2, 100)self.gridLayout.addLayout(self.horizontalLayout_2, 0, 0, 1, 1)self.stackedWidget.addWidget(self.page)self.page_2 = QtWidgets.QWidget()self.page_2.setObjectName("page_2")self.stackedWidget.addWidget(self.page_2)self.horizontalLayout.addWidget(self.stackedWidget)self.retranslateUi(widget)QtCore.QMetaObject.connectSlotsByName(widget)def retranslateUi(self, widget):_translate = QtCore.QCoreApplication.translatewidget.setWindowTitle(_translate("widget", "FaceCheck"))self.pushButton_start.setText(_translate("widget", "启动"))self.pushButton_cut.setText(_translate("widget", "截图"))self.pushButton_check_face.setText(_translate("widget", "识别人脸"))

# !/usr/bin/python3# -*- coding:utf-8 -*-"""@author: JHC000abc@gmail.com@file: main.py@time: 2023/8/6 16:31@desc: """from gui.ctrl.untitled_ctrl import UntitledCtrlimport osimport sysfrom PyQt5 import QtWidgetsfrom PyQt5.QtCore import QCoreApplication, Qtfrom qt_material import apply_stylesheetif __name__ == "__main__":root_path = os.getcwd()QCoreApplication.setAttribute(Qt.AA_EnableHighDpiScaling)app = QtWidgets.QApplication(sys.path)try:apply_stylesheet(app, theme="dark_teal.xml")except BaseException:print("样式加载失败")Form = UntitledCtrl(root_path)Form.show()sys.exit(app.exec_())

haarcascade_frontalface_default.xml
密码:JDBC

2