-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathMain_View_Qt.py
184 lines (136 loc) · 6.89 KB
/
Main_View_Qt.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
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
from PyQt5 import QtWidgets, uic
from PyQt5.QtWidgets import *
from PyQt5 import QtCore, QtGui
from PyQt5.QtGui import *
from PyQt5.QtCore import *
# 原生 Python 程式
import threading as THREAD
import time as TIME
import os as OS
import sys as SYSTEM
import datetime as DATETIME
# 頭部旋轉判斷程式
import head_rotation_detect as HEAD
from HistoryDict import *
# 取得ui檔案路徑
path = OS.getcwd()
# 設計好的ui檔案路徑
qtCreatorFile = path + OS.sep + "ui" + OS.sep + "mainView.ui"
# 讀入用Qt Designer設計的GUI layout
Ui_MainWindow, QtBaseClass = uic.loadUiType(qtCreatorFile)
# very testable class (hint: you can use mock.Mock for the signals)
class Worker(QtCore.QThread):
message = QtCore.pyqtSignal()
def __init__(self, parent=None):
super(Worker, self).__init__(parent)
self.working = True
def __del__(self):
self.working = False
self.wait()
def run(self):
while self.working == True:
# 發出訊號
self.message.emit()
TIME.sleep(0.08)
# Python的多重繼承 MainUi 繼承自兩個類別
class MainUi(QtWidgets.QMainWindow, Ui_MainWindow):
# =========================================================
# ======================= UI 主程式 ========================
def __init__(self):
QtWidgets.QMainWindow.__init__(self)
Ui_MainWindow.__init__(self)
self.setupUi(self)
# 取得內部物件
self.label_imageView = self.findChild(QLabel, name='imageView') # 圖片顯示 Label
self.label_fpsVal = self.findChild(QLabel, name='labelShow_FPS_Rate') # 顯示 FPS 的 Label
self.label_horizonVal = self.findChild(QLabel, name='labelShow_Angle_Horizon') # 顯示水平數值的 Label
self.label_verticalVal = self.findChild(QLabel, name='labelShow_Angle_Vertical') # 顯示垂直數值的 Label
self.label_headDirection = self.findChild(QLabel, name='labelShow_Head_Direction') # 目前頭部的轉向 Label
self.label_horizonThreshold = self.findChild(QLabel, name='labelText_Angle_Threshold_Horizon') # 顯示垂直閥值的 Label
self.label_verticalThreshold= self.findChild(QLabel, name='labelText_Angle_Threshold_Vertical') # 顯示垂直閥值的 Label
self.scrollBar_horizon = self.findChild(QSlider, name='sliderBar_Threshold_Horizon') # 水平閥值
self.scrollBar_vertical = self.findChild(QSlider, name='sliderBar_Threshold_Vertical') # 垂直閥值
self.label_focusRate = self.findChild(QLabel, name='labelShow_Focus_Rate') # 專注度
self.button_reset = self.findChild(QPushButton, name='pushButton_Reset')
self.checkBox_show3D = self.findChild(QCheckBox, name='checkBox_Show_3D')
self.checkBox_showFaceMark = self.findChild(QCheckBox, name='checkBox_Show_Facemark')
# 初始化影片顯示區域
self.initVideoEvent()
self.update_flag = True # 是否要繼續更新旗標
self.fps_rate = 0 # FPS
self.pre_time = TIME.time() # 取得目前時間
self.angle_history = History( 2, max_size=20 ) # 建立角度的紀錄
self.focus_count = 0
self.focus_timer = 0
def reset():
self.focus_count = 0
self.focus_timer = 0
self.button_reset.clicked.connect( reset )
# 建立新執行緒，將自定義訊號sinOut連線到slotAdd()槽函式
self.thread = Worker()
self.thread.message.connect(self.update)
self.thread.start()
def initVideoEvent(self):
pixmap = QPixmap('image.jpg')
self.label_imageView.setPixmap(pixmap)
def updateUI_second(self):
pass
# 更新影像函式
def update(self):
qApp.processEvents()
# 開始偵測
# QtWidgets.QApplication.instance().processEvents()
# ------------------------------------------------------------
show3D = self.checkBox_show3D.isChecked()
showMark = self.checkBox_showFaceMark.isChecked()
# 偵測頭部旋轉，並取得數值
(horizon_angle, vertical_angle, cvimg) = HEAD.detect( show3D, showMark )
try:
# 更新影像
image = QtGui.QImage(cvimg.data, cvimg.shape[1], cvimg.shape[0], QtGui.QImage.Format_RGB888).rgbSwapped()
image = QtGui.QPixmap.fromImage( image )
self.label_imageView.setPixmap(image)
except:
pass
# ------------------------------------------------------------
# 取得目前 ScrollBar 的值
hThreshold = self.scrollBar_horizon.value()
vThreshold = self.scrollBar_vertical.value()
self.label_horizonThreshold.setText('水平閥值： ' + str(hThreshold) )
self.label_verticalThreshold.setText('垂直閥值： ' + str(vThreshold) )
# 與之前的角度取平均，讓他更滑順
self.angle_history.add( [horizon_angle, vertical_angle] )
if( self.angle_history.is_full() ):
[horizon_angle, vertical_angle] = self.angle_history.get_average()
self.angle_history.pop()
# 更新頭部旋轉數值
self.label_horizonVal.setText( str( horizon_angle ) )
self.label_verticalVal.setText( str( vertical_angle ) )
# 取得目前頭部的轉向
(head_rotation, head_rotation_str) = HEAD.judge_look(horizon_angle =horizon_angle, vertical_angle =vertical_angle,
horizon_threshold =hThreshold , vertical_threshold=vThreshold)
rotation_str = '目前頭部方向： ' + str(head_rotation_str)
self.label_headDirection.setText( rotation_str )
# ------------------------------------------------------------
if head_rotation == HEAD.LEFT or head_rotation == HEAD.RIGHT or head_rotation == HEAD.GONE:
self.focus_count += 1
self.focus_timer += 1
self.label_focusRate.setText( str( round( (1 - (self.focus_count / self.focus_timer)) * 100, 2) ) + '%' )
# ------------------------------------------------------------
# 計算 FPS
self.fps_rate += 1
now_time = TIME.time()
if (now_time - self.pre_time) > 1:
self.updateUI_second()
self.label_fpsVal.setText( str( self.fps_rate ) )
self.fps_rate = 0
self.pre_time = now_time
# =========================================================
# ======================= 啟動程式 ========================
if __name__ == "__main__":
def run_app():
app = QtWidgets.QApplication( SYSTEM.argv )
window = MainUi()
window.show()
app.exec_()
run_app()