137 lines
4.2 KiB
Python
137 lines
4.2 KiB
Python
# -*- 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_())
|