diff --git a/MainWidget.py b/MainWidget.py new file mode 100644 index 0000000..510e583 --- /dev/null +++ b/MainWidget.py @@ -0,0 +1,184 @@ +from PyQt5.Qt import QWidget, QColor, QPixmap, QIcon, QSize, QCheckBox +from PyQt5.QtWidgets import QHBoxLayout, QVBoxLayout, QPushButton, QSplitter, QComboBox, QLabel, QSpinBox, QFileDialog,QTextEdit +from PyQt5.QtGui import QFont +from PaintBoard import PaintBoard +import numpy as np +# from model_core import LeNet5Custom +from model_core_win import LeNet5Custom + + +class MainWidget(QWidget): + def __init__(self, Parent=None): + super().__init__(Parent) + + # 先初始化数据，再初始化界面 + self.__InitData() + self.__InitView() + + + def __InitData(self): + self.__paintBoard = PaintBoard(self) + self.__colorList = QColor.colorNames() # 获取颜色列表(字符串类型) + + + def __InitView(self): + self.setFixedSize(896, 672) + self.setWindowTitle("Handwritten Digit Recognizer") + + # 新建一个水平布局作为本窗体的主布局 + main_layout = QHBoxLayout(self) + + # 设置主布局内边距以及控件间距为10px + main_layout.setSpacing(10) + + # 在主界面左侧放置画板 + main_layout.addWidget(self.__paintBoard) + + # 新建垂直子布局用于放置按键 + sub_layout = QVBoxLayout() + + # 设置此子布局和内部控件的间距为10px + sub_layout.setContentsMargins(10, 10, 10, 10) + + self.__btn_Clear = QPushButton("清空画板") + self.__btn_Clear.setParent(self) # 设置父对象为本界面 + self.__btn_Clear.clicked.connect(self.__paintBoard.Clear) # 将按键按下信号与画板清空函数相关联 + sub_layout.addWidget(self.__btn_Clear) + + self.__btn_predict = QPushButton("人工智能预测") + self.__btn_predict.setParent(self) # 设置父对象为本界面 + self.__btn_predict.clicked.connect(lambda:self.predict()) + sub_layout.addWidget(self.__btn_predict) + + self.__text_out = QTextEdit(self) + self.__text_out.setParent(self) + self.__text_out.setObjectName("预测结果为：") + self.__text_out.setReadOnly(True) + text_out_Font = QFont() + text_out_Font.setPointSize(20) + self.__text_out.setFont(text_out_Font) + sub_layout.addWidget(self.__text_out) + + self.__btn_Quit = QPushButton("退出") + self.__btn_Quit.setParent(self) # 设置父对象为本界面 + self.__btn_Quit.clicked.connect(self.Quit) + sub_layout.addWidget(self.__btn_Quit) + + self.__btn_Save = QPushButton("保存作品") + self.__btn_Save.setParent(self) + self.__btn_Save.clicked.connect(self.on_btn_Save_Clicked) + sub_layout.addWidget(self.__btn_Save) + + self.__btn_Load = QPushButton("加载作品") + self.__btn_Load.setParent(self) + self.__btn_Load.clicked.connect(self.on_btn_Load_Clicked) + sub_layout.addWidget(self.__btn_Load) + + self.__cbtn_Eraser = QCheckBox(" 使用橡皮擦") + self.__cbtn_Eraser.setParent(self) + self.__cbtn_Eraser.clicked.connect(self.on_cbtn_Eraser_clicked) + sub_layout.addWidget(self.__cbtn_Eraser) + + splitter = QSplitter(self) # 占位符 + sub_layout.addWidget(splitter) + + self.__label_penThickness = QLabel(self) + self.__label_penThickness.setText("画笔粗细") + self.__label_penThickness.setFixedHeight(20) + sub_layout.addWidget(self.__label_penThickness) + + self.__spinBox_penThickness = QSpinBox(self) + self.__spinBox_penThickness.setMaximum(50) # 规定粗细数值范围 + self.__spinBox_penThickness.setMinimum(1) + self.__spinBox_penThickness.setValue(20) # 默认粗细 + self.__spinBox_penThickness.setSingleStep(1) # 最小变化值 + self.__spinBox_penThickness.valueChanged.connect(self.on_PenThicknessChange) # 关联spinBox值变化信号和函数on_PenThicknessChange + sub_layout.addWidget(self.__spinBox_penThickness) + + self.__label_penColor = QLabel(self) + self.__label_penColor.setText("画笔颜色") + self.__label_penColor.setFixedHeight(20) + sub_layout.addWidget(self.__label_penColor) + + self.__comboBox_penColor = QComboBox(self) + self.__fillColorList(self.__comboBox_penColor) # 用各种颜色填充下拉列表 + self.__comboBox_penColor.currentIndexChanged.connect(self.on_PenColorChange) # 关联下拉列表的当前索引变更信号与函数 on_PenColorChange + sub_layout.addWidget(self.__comboBox_penColor) + + main_layout.addLayout(sub_layout) # 将子布局加入主布局 + + + def __fillColorList(self, comboBox): + index_black = 0 + index = 0 + for color in self.__colorList: + if color == "black": + index_black = index + index += 1 + pix = QPixmap(70, 20) + pix.fill(QColor(color)) + comboBox.addItem(QIcon(pix), None) + comboBox.setIconSize(QSize(70, 20)) + comboBox.setSizeAdjustPolicy(QComboBox.AdjustToContents) + + comboBox.setCurrentIndex(index_black) + + + def on_PenColorChange(self): + color_index = self.__comboBox_penColor.currentIndex() + color_str = self.__colorList[color_index] + self.__paintBoard.ChangePenColor(color_str) + + + def on_PenThicknessChange(self): + penThickness = self.__spinBox_penThickness.value() + self.__paintBoard.ChangePenThickness(penThickness) + + + def on_btn_Save_Clicked(self): + savePath = QFileDialog.getSaveFileName(self, 'Save Your Paint', '.\\', '*.png') + print(savePath) + if savePath[0] == "": + print("Save cancel") + return + image = self.__paintBoard.GetContentAsQImage() + image.save(savePath[0]) + + + def on_btn_Load_Clicked(self): + loadPath = QFileDialog.getOpenFileName(self, 'Load Your Paint', '.\\', '*.png') + print(loadPath) + if loadPath[0] == "": + print("Load cancel") + return + pixmap = QPixmap(loadPath[0]) + self.__paintBoard.SetContentFromQPixmap(pixmap) + + + def on_cbtn_Eraser_clicked(self): + if self.__cbtn_Eraser.isChecked(): + self.__paintBoard.EraserMode = True # 进入橡皮擦模式 + else: + self.__paintBoard.EraserMode = False # 退出橡皮擦模式 + + + def Quit(self): + self.close() + + + def predict(self): + savePath = './image_temp/test.png' + image = self.__paintBoard.GetContentAsQImage() + image.save(savePath) + + model = LeNet5Custom(dropout_rate=0.0) + model.load_model('./model/lenet5_model_best_20240621_010754.h5') + # lenet5_model_best_20240621_010754.h5 is best + + predictions = model.predict(savePath) + print(predictions) + predicted_class = np.argmax(predictions[0]) + + # 假设 self.__text_out 是一个文本输出控件 + self.__text_out.setText(str(predicted_class)) + print(f'Predictor: {predicted_class}')