# !/usr/bin/python3
# -*- coding:utf-8 -*-
"""
@author: JHC000abc@gmail.com
@file: untitled_ctrl.py
@time: 2023/8/6 16:29
@desc:

"""
import copy
import threading
import time
import traceback
import uuid
import os
import cv2
from gui.ui.untitled import Ui_widget
from PyQt5 import QtWidgets
from PyQt5.QtCore import pyqtSignal
from PyQt5.QtGui import QPixmap, QImage
from threading import Thread
from queue import Queue
import numpy
import PyQt5


class 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 = True
                            except 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, QtWidgets


class 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 UntitledCtrl
import os
import sys
from PyQt5 import QtWidgets
from PyQt5.QtCore import QCoreApplication, Qt
from qt_material import apply_stylesheet


if __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