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 = rootpath        self.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 = True        self.flag_face = True        self.cap = None        self.pause = False        self.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:                pass    def 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 = False        if 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:    pass                    image = 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("无法读取视频流")                    break                else:                    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, h        except BaseException:            pass    def start_pause(self):        """        :return:        """        if self.flag_start:            self.ui.pushButton_start.setText("暂停")            self.flag_start = False        else:            self.ui.pushButton_start.setText("开始")            self.flag_start = True    def face_check(self):        """        :return:        """        if self.flag_face:            self.flag_face = False            self.ui.pushButton_check_face.setText("暂停识别")        else:            self.flag_face = True            self.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.translate        widget.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

来源地址：https://blog.csdn.net/CXY00000/article/details/132137633