Add 'paintboard.py'

layout.py

@@ -0,0 +1,107 @@
+# -*- coding: utf-8 -*-
+
+# Form implementation generated from reading ui file 'layout.ui'
+#
+# Created by: PyQt5 UI code generator 5.14.1
+#
+# WARNING! All changes made in this file will be lost!
+
+
+from PyQt5 import QtCore, QtGui, QtWidgets
+
+
+class Ui_MainWindow(object):
+    def setupUi(self, MainWindow):
+        MainWindow.setObjectName("MainWindow")
+        MainWindow.resize(800, 600)
+        self.cbBox_Mode = QtWidgets.QComboBox(MainWindow)
+        self.cbBox_Mode.setGeometry(QtCore.QRect(60, 300, 211, 31))
+        self.cbBox_Mode.setObjectName("cbBox_Mode")
+        self.cbBox_Mode.addItem("")
+        self.cbBox_Mode.addItem("")
+        self.label = QtWidgets.QLabel(MainWindow)
+        self.label.setGeometry(QtCore.QRect(60, 270, 80, 20))
+        self.label.setObjectName("label")
+        self.pbtClear = QtWidgets.QPushButton(MainWindow)
+        self.pbtClear.setGeometry(QtCore.QRect(80, 440, 120, 30))
+        self.pbtClear.setStyleSheet("")
+        self.pbtClear.setCheckable(False)
+        self.pbtClear.setChecked(False)
+        self.pbtClear.setObjectName("pbtClear")
+        self.pbtGetMnist = QtWidgets.QPushButton(MainWindow)
+        self.pbtGetMnist.setGeometry(QtCore.QRect(80, 380, 120, 30))
+        self.pbtGetMnist.setCheckable(False)
+        self.pbtGetMnist.setObjectName("pbtGetMnist")
+        self.pbtPredict = QtWidgets.QPushButton(MainWindow)
+        self.pbtPredict.setGeometry(QtCore.QRect(80, 500, 120, 30))
+        self.pbtPredict.setStyleSheet("")
+        self.pbtPredict.setObjectName("pbtPredict")
+        self.lbDataArea = QtWidgets.QLabel(MainWindow)
+        self.lbDataArea.setGeometry(QtCore.QRect(540, 350, 224, 224))
+        self.lbDataArea.setMouseTracking(False)
+        self.lbDataArea.setStyleSheet("background-color: rgb(255, 255, 255);")
+        self.lbDataArea.setFrameShape(QtWidgets.QFrame.Box)
+        self.lbDataArea.setFrameShadow(QtWidgets.QFrame.Sunken)
+        self.lbDataArea.setLineWidth(4)
+        self.lbDataArea.setMidLineWidth(0)
+        self.lbDataArea.setText("")
+        self.lbDataArea.setObjectName("lbDataArea")
+        self.label_3 = QtWidgets.QLabel(MainWindow)
+        self.label_3.setGeometry(QtCore.QRect(260, 340, 91, 181))
+        self.label_3.setObjectName("label_3")
+        self.label_4 = QtWidgets.QLabel(MainWindow)
+        self.label_4.setGeometry(QtCore.QRect(540, 320, 131, 20))
+        self.label_4.setObjectName("label_4")
+        self.label_5 = QtWidgets.QLabel(MainWindow)
+        self.label_5.setGeometry(QtCore.QRect(20, 10, 711, 241))
+        self.label_5.setObjectName("label_5")
+        self.verticalLayoutWidget = QtWidgets.QWidget(MainWindow)
+        self.verticalLayoutWidget.setGeometry(QtCore.QRect(540, 350, 221, 221))
+        self.verticalLayoutWidget.setObjectName("verticalLayoutWidget")
+        self.dArea_Layout = QtWidgets.QVBoxLayout(self.verticalLayoutWidget)
+        self.dArea_Layout.setContentsMargins(0, 0, 0, 0)
+        self.dArea_Layout.setSpacing(0)
+        self.dArea_Layout.setObjectName("dArea_Layout")
+        self.lbResult = QtWidgets.QLabel(MainWindow)
+        self.lbResult.setGeometry(QtCore.QRect(380, 350, 91, 131))
+        font = QtGui.QFont()
+        font.setPointSize(48)
+        self.lbResult.setFont(font)
+        self.lbResult.setObjectName("lbResult")
+        self.lbCofidence = QtWidgets.QLabel(MainWindow)
+        self.lbCofidence.setGeometry(QtCore.QRect(360, 500, 151, 21))
+        font = QtGui.QFont()
+        font.setPointSize(12)
+        self.lbCofidence.setFont(font)
+        self.lbCofidence.setObjectName("lbCofidence")
+
+        self.retranslateUi(MainWindow)
+        self.cbBox_Mode.activated['QString'].connect(MainWindow.cbBox_Mode_Callback)
+        self.pbtClear.clicked.connect(MainWindow.pbtClear_Callback)
+        self.pbtPredict.clicked.connect(MainWindow.pbtPredict_Callback)
+        self.pbtGetMnist.clicked.connect(MainWindow.pbtGetMnist_Callback)
+        QtCore.QMetaObject.connectSlotsByName(MainWindow)
+
+    def retranslateUi(self, MainWindow):
+        _translate = QtCore.QCoreApplication.translate
+        MainWindow.setWindowTitle(_translate("MainWindow", "手写数字识别GUI-v1.0 --by hamlin"))
+        self.cbBox_Mode.setItemText(0, _translate("MainWindow", "1：MINIST随机抽取"))
+        self.cbBox_Mode.setItemText(1, _translate("MainWindow", "2：鼠标手写输入"))
+        self.label.setText(_translate("MainWindow", "模式选择"))
+        self.pbtClear.setText(_translate("MainWindow", "清除数据"))
+        self.pbtGetMnist.setText(_translate("MainWindow", "MNIST抽取"))
+        self.pbtPredict.setText(_translate("MainWindow", "识别"))
+        self.label_3.setText(_translate("MainWindow", "<html><head/><body><p><span style=\" font-size:12pt; font-weight:600;\">识别结果：</span></p><p><br/></p><p><br/></p><p><span style=\" font-size:12pt; font-weight:600;\">Softmax：</span></p></body></html>"))
+        self.label_4.setText(_translate("MainWindow", "数据输入区域"))
+        self.label_5.setText(_translate("MainWindow", "<!DOCTYPE HTML PUBLIC \"-//W3C//DTD HTML 4.0//EN\" \"http://www.w3.org/TR/REC-html40/strict.dtd\">\n"
+"<html><head><meta name=\"qrichtext\" content=\"1\" /><style type=\"text/css\">\n"
+"p, li { white-space: pre-wrap; }\n"
+"</style></head><body style=\" font-family:\'Ubuntu\'; font-size:11pt; font-weight:400; font-style:normal;\">\n"
+"<p style=\" margin-top:12px; margin-bottom:12px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;\"><span style=\" font-family:\'Ubuntu\'; font-weight:600;\">使用说明</span></p>\n"
+"<p style=\" margin-top:12px; margin-bottom:12px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;\"><span style=\" font-family:\'Ubuntu\';\">1、点击下拉列表进行模式选择，输入待识别数据后点击“识别”按键进行识别</span></p>\n"
+"<p style=\" margin-top:12px; margin-bottom:12px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;\"><span style=\" font-family:\'Ubuntu\';\">2、经CNN网络计算后输出，显示识别结果与Softmax值</span></p>\n"
+"<p style=\" margin-top:12px; margin-bottom:12px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;\"><span style=\" font-family:\'Ubuntu\';\">3、点击“清除数据”按键重新输入数据</span></p>\n"
+"<p style=\" margin-top:12px; margin-bottom:12px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;\"><span style=\" font-family:\'Ubuntu\';\">模式1：随机从测试集抽取图像作为待识别数据，点击“MNIST抽取”按键抽取</span></p>\n"
+"<p style=\" margin-top:12px; margin-bottom:12px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;\"><span style=\" font-family:\'Ubuntu\';\">模式2：使用鼠标在数据输入区域手写输入作为待识别数据</span></p></body></html>"))
+        self.lbResult.setText(_translate("MainWindow", "9"))
+        self.lbCofidence.setText(_translate("MainWindow", "0.99999999"))

paintboard.py

@@ -0,0 +1,82 @@
+#!/usr/bin/python3
+# -*- coding: utf-8 -*-
+
+import sys
+from PyQt5.QtWidgets import QWidget, QApplication
+from PyQt5.QtGui import QPixmap, QPainter, QPen, QColor
+from PyQt5.QtCore import Qt, QPoint, QSize
+
+class PaintBoard(QWidget):
+    def __init__(self, Parent = None, Size = QSize(320, 240), Fill = QColor(255,255,255,255)):
+        super().__init__(Parent)
+    
+        # 初始化参数
+        self.__size = Size                  # 画板尺寸
+        self.__fill = Fill                  # 画板默认填充颜色
+
+        self.__thickness = 18               # 默认画笔粗细
+        self.__penColor = QColor(0,0,0,255)   # 默认画笔颜色
+
+        self.__begin_point = QPoint()
+        self.__end_point = QPoint()
+
+        # 初始化画板界面
+        self.__board = QPixmap(self.__size)
+        self.__board.fill(Fill) 
+        self.setFixedSize(self.__size)
+        self.__painter = QPainter()         # 新建绘图工具
+
+
+    # 清空画板
+    def Clear(self):
+        self.__board.fill(self.__fill)
+        self.update()
+
+    def setBoardFill(self, fill):
+        self.__fill = fill
+        self.__board.fill(fill)
+        self.update()
+    
+    # 设置画笔颜色    
+    def setPenColor(self, color):
+        self.__penColor = color
+
+    # 设置画笔粗细    
+    def setPenThickness(self, thickness=10):    
+        self.__thickness = thickness  
+
+    # 获取画板QImage类型图片
+    def getContentAsQImage(self):
+        image = self.__board.toImage()
+        return image 
+
+    # 双缓冲绘图，绘图事件
+    def paintEvent(self, paintEvent):         
+        self.__painter.begin(self)
+        self.__painter.drawPixmap(0,0,self.__board)
+        self.__painter.end()
+
+    def mousePressEvent(self, mouseEvent):
+        if mouseEvent.button() == Qt.LeftButton:
+            self.__begin_point = mouseEvent.pos()
+            self.__end_point = self.__begin_point
+            # self.update()
+
+    def mouseMoveEvent(self, mouseEvent):
+        if mouseEvent.buttons() == Qt.LeftButton:
+            self.__end_point = mouseEvent.pos()
+
+            # 画入缓冲区
+            self.__painter.begin(self.__board)
+            self.__painter.setPen(QPen(self.__penColor,self.__thickness))  
+            self.__painter.drawLine(self.__begin_point, self.__end_point)
+            self.__painter.end()
+
+            self.__begin_point = self.__end_point
+            self.update()
+
+if __name__ == '__main__':
+    app = QApplication(sys.argv)
+    demo = PaintBoard()
+    demo.show()
+    sys.exit(app.exec_())