/
my_main_ui.py
136 lines (112 loc) · 4.22 KB
/
my_main_ui.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
# -*- coding: utf-8 -*-
"""
__author__ = xujing
__date__ = 2019-07-05
"""
from PyQt5.QtCore import pyqtSlot
from PyQt5.QtWidgets import QMainWindow
from PyQt5.QtCore import *
from PyQt5.QtWidgets import *
from PyQt5 import *
from PyQt5.QtGui import *
from PyQt5.QtCore import pyqtSlot
from PyQt5.QtWidgets import QMainWindow
from Ui_my_main_ui import Ui_MainWindow
import sys
import cv2
from car_id_detect import *
from svm_train import *
from card_seg import *
class MainWindow(QMainWindow, Ui_MainWindow):
"""
槽函数
"""
def __init__(self, parent=None):
"""
Constructor
@param parent reference to the parent widget
@type QWidget
"""
super(MainWindow, self).__init__(parent)
self.setupUi(self)
@pyqtSlot()
def on_pushButton_clicked(self):
"""
最下化
"""
print('最小化')
QMainWindow.showMinimized(self)
@pyqtSlot()
def on_pushButton_2_clicked(self):
"""
退出
"""
print("退出")
sys.exit(0)
@pyqtSlot()
def on_pushButton_6_clicked(self):
"""
加载图像
"""
print("加载图像")
try:
self.file_dir_temp,_ = QFileDialog.getOpenFileName(self,"选择被检测的车辆","D:/")
self.file_dir = self.file_dir_temp.replace("\\","/")
print(self.file_dir)
roi, label, color = CaridDetect(self.file_dir)
seg_dict, _, pre = Cardseg([roi],[color],None)
print(pre)
# segment
cv2.imwrite(os.path.join("./temp/seg_card.jpg"),roi)
seg_img = cv2.imread("./temp/seg_card.jpg")
seg_rows, seg_cols, seg_channels = seg_img.shape
bytesPerLine = seg_channels * seg_cols
cv2.cvtColor(seg_img, cv2.COLOR_BGR2RGB,seg_img)
QImg = QImage(seg_img.data, seg_cols, seg_rows,bytesPerLine, QImage.Format_RGB888)
self.label_2.setPixmap(QPixmap.fromImage(QImg).scaled(self.label_2.size(),
Qt.KeepAspectRatio, Qt.SmoothTransformation))
# reg result
pre.insert(2,"·")
self.label_3.setText(" "+"".join(pre))
# clor view
if color == "yello":
self.label_4.setStyleSheet("background-color: rgb(255, 255, 0);")
elif color == "green":
self.label_4.setStyleSheet("background-color: rgb(0, 255,0);")
elif color == "blue":
self.label_4.setStyleSheet("background-color: rgb(0, 0, 255);")
else:
self.label_4.setText("未识别出车牌颜色")
frame = cv2.imread(self.file_dir)
# cv2.rectangle(frame, (label[0],label[2]), (label[1],label[3]), (0,0,255), 2)
font = cv2.FONT_HERSHEY_SIMPLEX
cv2.putText(frame, 'https://github.com/DataXujing/vehicle-license-plate-recognition', (10, 10), font, 0.3, (0, 0, 255), 1)
img_rows, img_cols, channels = frame.shape
bytesPerLine = channels * img_cols
cv2.cvtColor(frame, cv2.COLOR_BGR2RGB,frame)
QImg = QImage(frame.data, img_cols, img_rows,bytesPerLine, QImage.Format_RGB888)
self.label.setPixmap(QPixmap.fromImage(QImg).scaled(self.label.size(),
Qt.KeepAspectRatio, Qt.SmoothTransformation))
QtWidgets.QApplication.processEvents()
except Exception as e:
QMessageBox.warning(self,"错误提示","[错误提示(请联系开发人员处理)]：\n" + str(e)+"\n或识别失败导致")
@pyqtSlot()
def on_pushButton_7_clicked(self):
"""
加载视频
"""
print("加载视频")
QMessageBox.information(self,"加载实时视频","未检测到实时视频源或暂未开通快该服务！")
if __name__ == "__main__":
app = QtWidgets.QApplication(sys.argv)
splash = QSplashScreen(QtGui.QPixmap(':/pic/pic/face.png'))
splash.show()
splash.showMessage('渲染界面...')
QThread.sleep(0.6)
splash.showMessage('正在初始化程序...')
QThread.sleep(0.6)
app. processEvents()
ui =MainWindow()
ui.show()
splash.finish(ui)
sys.exit(app.exec_())