-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmainwidget.py
159 lines (139 loc) · 6.79 KB
/
mainwidget.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
import os
import platform
from datetime import datetime
from PyQt5 import QtGui
from PyQt5.QtCore import Qt, pyqtSlot, QTimer, QSize, QRect
from PyQt5.QtGui import QImage, QPixmap, QPainter, QPen
from PyQt5 import QtWidgets
from PyQt5.QtGui import QMovie
# from play_audio import PlayAudioThread
# from serial_input import DetectorPersonThread
from ui.ui_main_widget import Ui_Form
from face_recognize import FaceRecognizeThread
from attend_record import AttendRecordThread
from loguru import logger
class MyWidget(QtWidgets.QWidget):
def __init__(self):
super().__init__()
self.ui = Ui_Form() # 实例化UI对象
self.ui.setupUi(self) # 初始化
# self.setWindowFlags(Qt.FramelessWindowHint)
# self.play_autio_thread = PlayAudioThread()
self.ui.widget.setVisible(False)
self.face_rec = FaceRecognizeThread(use_gpu=platform.system() != "Windows")
if platform.system() == "Windows":
self.connect_type = Qt.BlockingQueuedConnection
else:
self.connect_type = None
# self.people_rec = DetectorPersonThread()
# self.people_rec.people_entry_signal.connect(self.on_detective_people)
# self.people_rec.start()
# 实时画面传输信号
self.face_rec.img_finish_signal.connect(self.show_img, self.connect_type)
# 打卡信息保存信号
self.face_rec.record_attend_signal.connect(self.delt, self.connect_type)
# 识别结果信号-触发显示信息
self.face_rec.rec_result_signal.connect(self.change_sign_info, self.connect_type)
# 判断是否有人脸的信号--触发控件 visible
self.face_rec.is_person_signal.connect(self.set_cover_invisible, self.connect_type)
# 绘制矩形框
self.face_rec.det_signal.connect(self.draw_rect, self.connect_type)
QTimer.singleShot(500, self.face_rec.start)
# 显示字体大小
self._rect = None
self.cur_font_size = 25
self.font_size_rate = self.cur_font_size / self.geometry().height()
# 时间显示定时器
timer = QTimer(self)
timer.timeout.connect(lambda: self.ui.label_time.setText(datetime.now().strftime("%Y-%m-%d %H:%M:%S")))
timer.start(1000)
# 设置人脸区域图像
# pix_map = QPixmap("./resource/images/face_region.png")
# self.ui.label.setPixmap(pix_map.scaled(320, pix_map.height(), Qt.KeepAspectRatio))
# move = QMovie("./resource/images/face_region.gif")
# move.setScaledSize(QSize(600, 600))
# move.start()
# self.ui.label.setMovie(move)
# op = QtWidgets.QGraphicsOpacityEffect()
# # 设置透明度的值，0.0到1.0，最小值0是透明，1是不透明
# op.setOpacity(0.5)
# self.ui.label.setGraphicsEffect(op)
@pyqtSlot(QImage)
def show_img(self, qimg):
if self._rect is not None:
painter = QPainter(qimg)
pen = QPen(Qt.yellow, 2, Qt.SolidLine)
painter.setPen(pen)
painter.drawRect(self._rect[0], self._rect[1], self._rect[2], self._rect[3])
scaled_img = qimg.scaled(self.ui.image_label.size(),
Qt.KeepAspectRatio,
Qt.SmoothTransformation)
self.ui.image_label.setPixmap(QPixmap.fromImage(scaled_img))
def closeEvent(self, event: QtGui.QCloseEvent) -> None:
if self.face_rec.isRunning():
self.face_rec.finished()
self.face_rec.wait()
@pyqtSlot(list)
def draw_rect(self, rect: list):
self._rect = rect
@pyqtSlot(list)
def delt(self, records: list):
# calculate filename
th = AttendRecordThread(records)
th.start()
th.wait()
def change_sign_info(self, info: dict):
if info["code"] is None:
return
self.ui.sign_time.setText(info["time"])
# 设置文字
if info["code"] == -1:
self.ui.sign_name.setText(info["message"])
self.ui.cover.setStyleSheet("background-color:rgba(255, 0, 0, 40)")
if info["code"] == 1:
self.ui.sign_name.setText(info["message"].face_info)
# 设置样式
self.ui.sign_name.setStyleSheet(f"font-size:{self.cur_font_size}pt; font-weight:600;color:#0dc839;")
self.ui.sign_time.setStyleSheet(f"font-size:{self.cur_font_size}pt; font-weight:600;color:#0dc839;")
self.ui.widget.setStyleSheet("#widget{background-color: rgba(46, 255, 0, 40);}")
self.ui.cover.setStyleSheet("background-color:rgba(255, 0, 0, 0)")
self.ui.label_attend_img.setPixmap(QPixmap("./resource/images/signin_success.png"))
# if not self.play_autio_thread.isRunning():
# self.play_autio_thread.start()
if info["code"] == 0:
self.ui.sign_name.setText(info["message"])
self.ui.sign_name.setStyleSheet(f"font-size:{self.cur_font_size}pt; font-weight:600;color:red;")
self.ui.sign_time.setStyleSheet(f"font-size:{self.cur_font_size}pt; font-weight:600;color:red;")
self.ui.widget.setStyleSheet("#widget{background-color: rgba(255, 0, 0, 40);}")
self.ui.label_attend_img.setPixmap(QPixmap("./resource/images/signin_fail.png"))
self.ui.cover.setStyleSheet("background-color:rgba(255, 0, 0, 0)")
if not self.ui.widget.isVisible():
self.ui.widget.setVisible(True)
def set_cover_invisible(self):
if self.ui.widget.isVisible():
self.ui.widget.setVisible(False)
self.ui.cover.setStyleSheet("background-color:rgba(255, 0, 0, 0)")
if self._rect is not None:
self._rect = None
def paintEvent(self, a0: QtGui.QPaintEvent) -> None:
if not self.isActiveWindow():
self.activateWindow()
def resizeEvent(self, a0: QtGui.QResizeEvent) -> None:
self.ui.image_label.resize(self.geometry().width(), self.geometry().height())
cur_height_ = self.geometry().height()
cur_with_ = self.geometry().width()
#
# self.ui.cover.move(0, 0)
self.ui.cover.resize(cur_with_, cur_height_)
self.cur_font_size = self.font_size_rate * self.geometry().height()
self.ui.label_1.setStyleSheet(f"font-size:{self.cur_font_size}pt; font-weight:600;color:#ffffff;")
self.ui.label_2.setStyleSheet(f"font-size:{self.cur_font_size}pt; font-weight:600;color:#ffffff;")
self.ui.label_time.setStyleSheet(f"font-size:{self.cur_font_size}pt; font-weight:600;color:rgb(85, 85, 85);")
def on_detective_people(self):
self.face_rec.start_tracker()
def keyPressEvent(self, a0: QtGui.QKeyEvent) -> None:
if a0.key() == Qt.Key_F:
if not self.isFullScreen():
self.showFullScreen()
else:
self.showNormal()