From 5dc25bb11b44c9cad6291a88c24290d3215ba2f5 Mon Sep 17 00:00:00 2001 From: pgw247brm Date: Sat, 1 Jul 2023 22:15:57 +0800 Subject: [PATCH] ADD file via upload --- main.py | 468 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 468 insertions(+) create mode 100644 main.py diff --git a/main.py b/main.py new file mode 100644 index 0000000..d7813ed --- /dev/null +++ b/main.py @@ -0,0 +1,468 @@ +import cv2 +import numpy as np +import PyQt5.QtGui +from PyQt5.QtGui import QPixmap, QImage, qRgb + +import Mainwindow +from PyQt5.QtWidgets import * +import sys + +import picture_fun + + +class MainWindow(QWidget,Mainwindow.Ui_Mainwindow): + def __init__(self): + super(MainWindow, self).__init__() + self.setupUi(self) + + self.stackedLayout = QStackedLayout(self.frame) + + self.page1 = FramePage1() + self.page2 = FramePage2() + self.page3 = FramePage3() + self.page4 = FramePage4() + self.page5 = FramePage5() + self.page6 = FramePage6() + self.page7 = FramePage7() + + self.stackedLayout.addWidget(self.page1) #添加控件 + self.stackedLayout.addWidget(self.page2) + self.stackedLayout.addWidget(self.page3) + self.stackedLayout.addWidget(self.page4) + self.stackedLayout.addWidget(self.page5) + self.stackedLayout.addWidget(self.page6) + self.stackedLayout.addWidget(self.page7) + + self.controller() + + def controller(self): + self.pushButton_1.clicked.connect(self.switch1) + self.pushButton_2.clicked.connect(self.switch2) + self.pushButton_3.clicked.connect(self.switch3) + self.pushButton_4.clicked.connect(self.switch4) + self.pushButton_5.clicked.connect(self.switch5) + self.pushButton_6.clicked.connect(self.switch6) + self.pushButton_7.clicked.connect(self.switch7) + + def switch1(self): + self.stackedLayout.setCurrentIndex(0) # 索引按加入布局的顺序 + def switch2(self): + self.stackedLayout.setCurrentIndex(1) + def switch3(self): + self.stackedLayout.setCurrentIndex(2) + def switch4(self): + self.stackedLayout.setCurrentIndex(3) + def switch5(self): + self.stackedLayout.setCurrentIndex(4) + def switch6(self): + self.stackedLayout.setCurrentIndex(5) + def switch7(self): + self.stackedLayout.setCurrentIndex(6) + + + + + +class FramePage1(QWidget, Mainwindow.Ui_Formwin1): + def __init__(self): + super().__init__() + self.setupUi(self) + self.pushButton_load.clicked.connect(self.openimage) + self.pushButton_save.clicked.connect(self.saveimage) + self.pushButton_turntoGray.clicked.connect(self.fun1) + self.pushButton_turntotwo.clicked.connect(self.fun2) + + def openimage(self): + fname = QFileDialog.getOpenFileName(self, '打开图片', './', "Images (*.png *.jpg *.bmp)") + if fname[0]: + self.label_daichuli.setPixmap(QPixmap(fname[0])) + self.label_daichuli.setWordWrap(True) + self.label_daichuli.setScaledContents(True) + + def saveimage(self): + nfname = QFileDialog.getSaveFileName(self, "保存图片", "./", "Images (*.png *.jpg *.bmp)") + if nfname[0]: + self.label_jieguo.pixmap().save(nfname[0]) + + def fun1(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + newsrc = picture_fun.gray_picture(src) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + def fun2(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + newsrc = picture_fun.erzhihua(src) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + + +class FramePage2(QWidget, Mainwindow.Ui_Formwin2): + def __init__(self): + super().__init__() + self.setupUi(self) + self.pushButton_load.clicked.connect(self.openimage) + self.pushButton_save.clicked.connect(self.saveimage) + self.pushButton_fun0.clicked.connect(self.fun1) + self.pushButton_fun1.clicked.connect(self.fun2) + self.pushButton_fun2.clicked.connect(self.fun3) + self.pushButton_fun3.clicked.connect(self.fun4) + + def openimage(self): + fname = QFileDialog.getOpenFileName(self, '打开图片', './', "Images (*.png *.jpg *.bmp *.jpeg)") + if fname[0]: + self.label_daichuli.setPixmap(QPixmap(fname[0])) + self.label_daichuli.setWordWrap(True) + self.label_daichuli.setScaledContents(True) + + def saveimage(self): + nfname = QFileDialog.getSaveFileName(self, "保存图片", "./", "Images (*.png *.jpg *.bmp)") + if nfname[0]: + self.label_jieguo.pixmap().save(nfname[0]) + + def fun1(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + newsrc = picture_fun.flipfun(src,0) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + def fun2(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + newsrc = picture_fun.flipfun(src,1) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + def fun3(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + newsrc = picture_fun.flipfun(src,-1) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + def fun4(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + x = self.lineEdit_power_value.text() + x = float(x) + newsrc = picture_fun.fangsuo(src, x) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + +class FramePage3(QWidget, Mainwindow.Ui_Formwin3): + def __init__(self): + super().__init__() + self.setupUi(self) + self.pushButton_load.clicked.connect(self.openimage) + self.pushButton_save.clicked.connect(self.saveimage) + self.pushButton_lap.clicked.connect(self.lap_fun) + self.pushButton_zhifangtu.clicked.connect(self.equalhist_fun) + self.pushButton_gama.clicked.connect(self.gama_fun) + + + def openimage(self): + fname = QFileDialog.getOpenFileName(self, '打开图片', './', "Images (*.png *.jpg *.bmp)") + if fname[0]: + self.label_daichuli.setPixmap(QPixmap(fname[0])) + self.label_daichuli.setWordWrap(True) + self.label_daichuli.setScaledContents(True) + + def saveimage(self): + nfname = QFileDialog.getSaveFileName(self, "保存图片", "./", "Images (*.png *.jpg *.bmp)") + if nfname[0]: + self.label_jieguo.pixmap().save(nfname[0]) + + def lap_fun(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + newsrc = picture_fun.lap_9(src) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + def equalhist_fun(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + newsrc = picture_fun.cal_equalhist(src) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + def gama_fun(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + power = self.lineEdit_power_value.text() + power = float(power) + newsrc = picture_fun.gama_transfer(src, power) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + +class FramePage4(QWidget, Mainwindow.Ui_Formwin4): + def __init__(self): + super().__init__() + self.setupUi(self) + self.pushButton_load.clicked.connect(self.openimage) + self.pushButton_save.clicked.connect(self.saveimage) + self.pushButton_salt.clicked.connect(self.add_noisy_fun) #椒盐彩色 + self.pushButton_gauss.clicked.connect(self.add_noise_fun) #彩色图像添加高斯噪声 + + def openimage(self): + fname = QFileDialog.getOpenFileName(self, '打开图片', './', "Images (*.png *.jpg *.bmp)") + if fname[0]: + self.label_daichuli.setPixmap(QPixmap(fname[0])) + self.label_daichuli.setWordWrap(True) + self.label_daichuli.setScaledContents(True) + + def saveimage(self): + nfname = QFileDialog.getSaveFileName(self, "保存图片", "./", "Images (*.png *.jpg *.bmp)") + if nfname[0]: + self.label_jieguo.pixmap().save(nfname[0]) + + def add_noisy_fun(self): #椒盐彩色 + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + num = self.lineEdit_n_value.text() + num= float(num) + newsrc = picture_fun.add_noisy(src, num) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + def add_noise_fun(self): #彩色图像添加高斯噪声 + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + mean = self.lineEdit_mean_value.text() + mean = int(mean) + val=self.lineEdit_val_value.text() + val=float(val) + newsrc = picture_fun.add_noise(src,mean,val) + pix = matqimage_guass(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + +class FramePage5(QWidget, Mainwindow.Ui_Formwin5): + def __init__(self): + super().__init__() + self.setupUi(self) + self.pushButton_load.clicked.connect(self.openimage) + self.pushButton_save.clicked.connect(self.saveimage) + self.pushButton_fangbo.clicked.connect(self.boxFilter_fun) + self.pushButton_zhongzhi.clicked.connect(self.medianBlur_fun) + self.pushButton_suangbian.clicked.connect(self.bilateralFilter_fun) + self.pushButton_gauss.clicked.connect(self.GaussianBlur_fun) + self.pushButton_junzhi.clicked.connect(self.blur_fun) + + def openimage(self): + fname = QFileDialog.getOpenFileName(self, '打开图片', './', "Images (*.png *.jpg *.bmp)") + if fname[0]: + self.label_daichuli.setPixmap(QPixmap(fname[0])) + self.label_daichuli.setWordWrap(True) + self.label_daichuli.setScaledContents(True) + + def saveimage(self): + nfname = QFileDialog.getSaveFileName(self, "保存图片", "./", "Images (*.png *.jpg *.bmp)") + if nfname[0]: + self.label_jieguo.pixmap().save(nfname[0]) + + def boxFilter_fun(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + newsrc = picture_fun.boxFilterfun(src) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + def medianBlur_fun(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + newsrc = picture_fun.medianBlurfun(src) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + def bilateralFilter_fun(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + newsrc = picture_fun.bilateralFilterfun(src) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + def GaussianBlur_fun(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + newsrc = picture_fun.GaussianBlurfun(src) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + def blur_fun(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + newsrc = picture_fun.blurfun(src) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + +class FramePage6(QWidget, Mainwindow.Ui_Formwin6): + def __init__(self): + super().__init__() + self.setupUi(self) + self.pushButton_load.clicked.connect(self.openimage) + self.pushButton_save.clicked.connect(self.saveimage) + self.pushButton_sift.clicked.connect(self.siftfun) + self.pushButton_lunkuo.clicked.connect(self.lunkuofun) + + def openimage(self): + fname = QFileDialog.getOpenFileName(self, '打开图片', './', "Images (*.png *.jpg *.bmp)") + if fname[0]: + self.label_daichuli.setPixmap(QPixmap(fname[0])) + self.label_daichuli.setWordWrap(True) + self.label_daichuli.setScaledContents(True) + + def saveimage(self): + nfname = QFileDialog.getSaveFileName(self, "保存图片", "./", "Images (*.png *.jpg *.bmp)") + if nfname[0]: + self.label_jieguo.pixmap().save(nfname[0]) + + + def siftfun(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + newsrc = picture_fun.sift_fun(src) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + def lunkuofun(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + newsrc = picture_fun.morphologyExfun(src) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) + + +class FramePage7(QWidget, Mainwindow.Ui_Formwin7): + def __init__(self): + super().__init__() + self.setupUi(self) + self.pushButton_load.clicked.connect(self.openimage) + self.pushButton_save.clicked.connect(self.saveimage) + self.pushButton_xiufu.clicked.connect(self.xiufu_fun) + + def openimage(self): + fname = QFileDialog.getOpenFileName(self, '打开图片', './', "Images (*.png *.jpg *.bmp)") + if fname[0]: + self.label_daichuli.setPixmap(QPixmap(fname[0])) + self.label_daichuli.setWordWrap(True) + self.label_daichuli.setScaledContents(True) + + def saveimage(self): + nfname = QFileDialog.getSaveFileName(self, "保存图片", "./", "Images (*.png *.jpg *.bmp)") + if nfname[0]: + self.label_jieguo.pixmap().save(nfname[0]) + + def xiufu_fun(self): + qimg = self.label_daichuli.pixmap() + src = qimage2mat(qimg) + newsrc = picture_fun.xiufu(src) + pix = matqimage(newsrc) + self.label_jieguo.setPixmap(pix) + self.label_jieguo.setWordWrap(True) + self.label_jieguo.setScaledContents(True) +############################################################################################################# + +def qimage2mat(qtpixmap): #qtpixmap转opencv + qimg = qtpixmap.toImage() + temp_shape = (qimg.height(), qimg.bytesPerLine() * 8 // qimg.depth()) + temp_shape += (4,) + ptr = qimg.bits() + ptr.setsize(qimg.byteCount()) + result = np.array(ptr, dtype=np.uint8).reshape(temp_shape) + result = result[..., :3] + return result + + +def matqimage(cvimg): #opencv转QImage + if cvimg.ndim==2: #单通道 + height, width= cvimg.shape + cvimg = cv2.cvtColor(cvimg, cv2.COLOR_BGR2RGB) + cvimg = QImage(cvimg.data, width, height, QImage.Format_RGB888) + pix = QPixmap.fromImage(cvimg) + return pix + else: #多个通道 + width = cvimg.shape[1] + height = cvimg.shape[0] + pixmap = QPixmap(width, height) # 根据已知的高度和宽度新建一个空的QPixmap, + qimg = pixmap.toImage() # 将pximap转换为QImage类型的qimg + for row in range(0, height): + for col in range(0, width): + b = cvimg[row, col, 0] + g = cvimg[row, col, 1] + r = cvimg[row, col, 2] + pix = qRgb(r, g, b) + qimg.setPixel(col, row, pix) + pix = QPixmap.fromImage(qimg) + return pix + +def matqimage_guass(cvimg): #opencv转QImage #guass 专用#### + if cvimg.ndim==2: #单通道 + height, width= cvimg.shape + cvimg = cv2.cvtColor(cvimg, cv2.COLOR_BGR2RGB) + cvimg = QImage(cvimg.data, width, height, QImage.Format_RGB888) + pix = QPixmap.fromImage(cvimg) + return pix + else: #多个通道 + width = cvimg.shape[1] # 获取图片宽度 + height = cvimg.shape[0] # 获取图片高度 + pixmap = QPixmap(width, height) # 根据已知的高度和宽度新建一个空的QPixmap, + qimg = pixmap.toImage() # 将pximap转换为QImage类型的qimg + for row in range(0, height): + for col in range(0, width): + b = int(cvimg[row, col, 0]*255) #高斯加噪归一化了 要*255 + g = int(cvimg[row, col, 1]*255) + r = int(cvimg[row, col, 2]*255) + pix = qRgb(r, g, b) + qimg.setPixel(col, row, pix) + pix = QPixmap.fromImage(qimg) + return pix # 转换完成，返回 + +if __name__ == '__main__': + app = QApplication(sys.argv) + ui = MainWindow() + ui.show() + sys.exit(app.exec_()) \ No newline at end of file