5-31

mainEntry.py

@@ -1,7 +1,6 @@
 import os
 import sys
 import cv2 as cv
-import numpy as np
 
 from PyQt5 import QtCore, QtGui, QtWidgets
 from PyQt5.QtCore import *
@@ -13,52 +12,70 @@ os.chdir(sys.path[0])
 
 class PyQtMainEntry(QMainWindow, Ui_MainWindow):
     def __init__(self):
-        super().__init__()
-        self.setupUi(self)
 
+        super().__init__()  # 调用父类的初始化函数
+        self.setupUi(self)  # 设置用户界面
+
+        # 初始化摄像头，并尝试打开，默认尝试打开摄像头索引为0
         self.camera = cv.VideoCapture(0)
-        self.is_camera_opened = False
+        self.is_camera_opened = False  # 标记摄像头是否成功打开
 
+        # 创建一个定时器，并设置其触发间隔为10毫秒
         self._timer = QtCore.QTimer(self)
-        self._timer.timeout.connect(self._queryFrame)
+        self._timer.timeout.connect(self._queryFrame)  # 定时器触发时调用_queryFrame函数
-        self._timer.setInterval(10)
+        self._timer.setInterval(10)  # 设置定时器触发间隔
 
-    def btnOpenCamera_Clicked(self):
 
+    def btnOpenCamera_Clicked(self):
 
+        # 切换摄像头的打开状态
         self.is_camera_opened = ~self.is_camera_opened
         if self.is_camera_opened:
+            # 当摄像头打开时，更改按钮文本为关闭摄像头，并启动定时器
             self.btnOpenCamera.setText("关闭摄像头")
             self._timer.start()
         else:
+            # 当摄像头关闭时，更改按钮文本为打开摄像头，并停止定时器
             self.btnOpenCamera.setText("打开摄像头")
             self._timer.stop()
 
     def btnCapture_Clicked(self):
 
         if not self.is_camera_opened:
+            # 如果相机未打开，则直接返回，不做任何操作
             return
-
+        # 捕获图像
         self.captured = self.frame
 
+        # 获取捕获图像的行数、列数和通道数
         rows, cols, channels = self.captured.shape
+        # 计算每行字节数
         bytesPerLine = channels * cols
 
+        # 使用捕获的图像数据创建QImage对象
         QImg = QImage(self.captured.data, cols, rows, bytesPerLine, QImage.Format_RGB888)
+        # 将QImage对象转换为QPixmap，并按比例缩放以适应标签的大小，然后设置到标签上
         self.labelCapture.setPixmap(QPixmap.fromImage(QImg).scaled(
             self.labelCapture.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation))
 
     def btnReadImage_Clicked(self):
 
+        # 使用QFileDialog获取选择的图像文件名
+        # ,_是占位符用来忽略返回值比如这里面会返回文件名和文件类型，使用占位符可以忽略文件类型
         filename,  _ = QFileDialog.getOpenFileName(self, '打开图片')
         if filename:
+            # 通过文件名读取图像，并转换为RGB格式
             self.captured = cv.imread(str(filename))
-
             self.captured = cv.cvtColor(self.captured, cv.COLOR_BGR2RGB)
 
+            # 获取图像的尺寸信息，用于创建QImage
             rows, cols, channels = self.captured.shape
             bytesPerLine = channels * cols
+
+            # 将OpenCV图像转换为QImage格式
             QImg = QImage(self.captured.data, cols, rows, bytesPerLine, QImage.Format_RGB888)
+
+            # 将QImage设置为标签的图片，并保持原始比例进行缩放
             self.labelCapture.setPixmap(QPixmap.fromImage(QImg).scaled(
                 self.labelCapture.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation))
 
@@ -93,19 +110,27 @@ class PyQtMainEntry(QMainWindow, Ui_MainWindow):
 
     @QtCore.pyqtSlot()
     def _queryFrame(self):
-
+        """
+        从相机读取一帧图像，并将其显示在标签上。
+        """
+        # 从相机读取图像
         ret, self.frame = self.camera.read()
 
+        # 获取图像的尺寸和每行字节数
         img_rows, img_cols, channels = self.frame.shape
         bytesPerLine = channels * img_cols
 
+        # 将图像从BGR转换为RGB格式
         cv.cvtColor(self.frame, cv.COLOR_BGR2RGB, self.frame)
+        # 创建QImage对象
         QImg = QImage(self.frame.data, img_cols, img_rows, bytesPerLine, QImage.Format_RGB888)
+        # 将QImage设置为标签的图片，并保持宽高比
         self.labelCamera.setPixmap(QPixmap.fromImage(QImg).scaled(
             self.labelCamera.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation))
 
 
 
+
     def btnDenoising_Clicked(self):
         if not hasattr(self, "captured"):
             return
@@ -119,6 +144,17 @@ class PyQtMainEntry(QMainWindow, Ui_MainWindow):
         self.labelResult.setPixmap(QPixmap.fromImage(QImg).scaled(
             self.labelResult.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation))
 
+
+    def btnSaveImage_Clicked(self):
+
+        if not hasattr(self, "captured"):  # 检查是否已捕获图像
+            return
+
+        filename,  _ = QFileDialog.getSaveFileName(self, '保存图片')  # 弹出保存文件对话框
+        if filename:  # 如果用户选择了保存路径
+            cv.imwrite(str(filename), self.captured)  # 将捕获的图像保存到选择的路径
+
+
 if __name__ == "__main__":
     app = QtWidgets.QApplication(sys.argv)
     window = PyQtMainEntry()

surface.py

@@ -14,7 +14,7 @@ from PyQt5 import QtCore, QtGui, QtWidgets
 class Ui_MainWindow(object):
     def setupUi(self, MainWindow):
         MainWindow.setObjectName("MainWindow")
-        MainWindow.resize(832, 441)
+        MainWindow.resize(664, 436)
         self.centralwidget = QtWidgets.QWidget(MainWindow)
         self.centralwidget.setObjectName("centralwidget")
         self.btnOpenCamera = QtWidgets.QPushButton(self.centralwidget)
@@ -33,21 +33,24 @@ class Ui_MainWindow(object):
         self.btnGray.setGeometry(QtCore.QRect(190, 310, 75, 23))
         self.btnGray.setObjectName("btnGray")
         self.labelCamera = QtWidgets.QLabel(self.centralwidget)
-        self.labelCamera.setGeometry(QtCore.QRect(0, 40, 150, 150))
+        self.labelCamera.setGeometry(QtCore.QRect(20, 40, 150, 150))
         self.labelCamera.setFrameShadow(QtWidgets.QFrame.Plain)
         self.labelCamera.setObjectName("labelCamera")
         self.labelCapture = QtWidgets.QLabel(self.centralwidget)
-        self.labelCapture.setGeometry(QtCore.QRect(150, 40, 150, 150))
+        self.labelCapture.setGeometry(QtCore.QRect(200, 40, 150, 150))
         self.labelCapture.setObjectName("labelCapture")
         self.labelResult = QtWidgets.QLabel(self.centralwidget)
-        self.labelResult.setGeometry(QtCore.QRect(310, 40, 150, 150))
+        self.labelResult.setGeometry(QtCore.QRect(380, 40, 150, 150))
         self.labelResult.setObjectName("labelResult")
         self.btnDenoising = QtWidgets.QPushButton(self.centralwidget)
         self.btnDenoising.setGeometry(QtCore.QRect(330, 310, 75, 23))
         self.btnDenoising.setObjectName("btnDenoising")
+        self.btnSaveImage = QtWidgets.QPushButton(self.centralwidget)
+        self.btnSaveImage.setGeometry(QtCore.QRect(190, 360, 75, 23))
+        self.btnSaveImage.setObjectName("btnSaveImage")
         MainWindow.setCentralWidget(self.centralwidget)
         self.menubar = QtWidgets.QMenuBar(MainWindow)
-        self.menubar.setGeometry(QtCore.QRect(0, 0, 832, 22))
+        self.menubar.setGeometry(QtCore.QRect(0, 0, 664, 22))
         self.menubar.setObjectName("menubar")
         MainWindow.setMenuBar(self.menubar)
         self.statusbar = QtWidgets.QStatusBar(MainWindow)
@@ -61,6 +64,7 @@ class Ui_MainWindow(object):
         self.btnThreshold.clicked.connect(MainWindow.btnThreshold_Clicked) # type: ignore
         self.btnGray.clicked.connect(MainWindow.btnGray_Clicked) # type: ignore
         self.btnDenoising.clicked.connect(MainWindow.btnDenoising_Clicked) # type: ignore
+        self.btnSaveImage.clicked.connect(MainWindow.btnSaveImage_Clicked) # type: ignore
         QtCore.QMetaObject.connectSlotsByName(MainWindow)
         MainWindow.setTabOrder(self.btnOpenCamera, self.btnCapture)
         MainWindow.setTabOrder(self.btnCapture, self.btnReadImage)
@@ -79,3 +83,4 @@ class Ui_MainWindow(object):
         self.labelCapture.setText(_translate("MainWindow", "捕获图"))
         self.labelResult.setText(_translate("MainWindow", "结果图"))
         self.btnDenoising.setText(_translate("MainWindow", "加噪"))
+        self.btnSaveImage.setText(_translate("MainWindow", "保存图片"))