ADD file via upload

styleTransfer.py

@@ -0,0 +1,165 @@
+import sys
+import cv2
+import numpy as np
+from PySide2.QtCore import *
+from PySide2.QtGui import *
+from PySide2.QtWidgets import *
+
+class Ui_MainWindow(object):
+    def setupUi(self, MainWindow):
+        if not MainWindow.objectName():
+            MainWindow.setObjectName(u"MainWindow")
+        MainWindow.resize(1000, 800)
+        self.centralwidget = QWidget(MainWindow)
+        self.centralwidget.setObjectName(u"centralwidget")
+        self.pushButton = QPushButton(self.centralwidget)
+        self.pushButton.setObjectName(u"pushButton")
+        self.pushButton.setGeometry(QRect(750, 500, 201, 71))
+        self.graphicsView = QGraphicsView(self.centralwidget)
+        self.graphicsView.setObjectName(u"graphicsView")
+        self.graphicsView.setGeometry(QRect(90, 80, 401, 400))
+        self.pushButton_2 = QPushButton(self.centralwidget)
+        self.pushButton_2.setObjectName(u"pushButton_2")
+        self.pushButton_2.setGeometry(QRect(100, 500, 201, 71))
+        self.graphicsView_2 = QGraphicsView(self.centralwidget)
+        self.graphicsView_2.setObjectName(u"graphicsView_2")
+        self.graphicsView_2.setGeometry(QRect(520, 80, 401, 400))
+
+        self.comboBox = QComboBox(self.centralwidget)
+        self.comboBox.addItem("")
+        self.comboBox.addItem("")
+        self.comboBox.addItem("")
+        self.comboBox.addItem("")
+
+        self.comboBox.addItem("")
+        self.comboBox.addItem("")
+        self.comboBox.addItem("")
+        self.comboBox.addItem("")
+
+        self.comboBox.setObjectName(u"comboBox")
+        self.comboBox.setGeometry(QRect(150, 620, 101, 31))
+        self.label = QLabel(self.centralwidget)
+        self.label.setObjectName(u"label")
+        self.label.setGeometry(QRect(150, 580, 200, 31))
+        self.pushButton_3 = QPushButton(self.centralwidget)
+        self.pushButton_3.setObjectName(u"pushButton_3")
+        self.pushButton_3.setGeometry(QRect(750, 600, 201, 71))
+        MainWindow.setCentralWidget(self.centralwidget)
+        self.menubar = QMenuBar(MainWindow)
+        self.menubar.setObjectName(u"menubar")
+        self.menubar.setGeometry(QRect(0, 0, 1000, 23))
+        MainWindow.setMenuBar(self.menubar)
+        self.statusbar = QStatusBar(MainWindow)
+        self.statusbar.setObjectName(u"statusbar")
+        MainWindow.setStatusBar(self.statusbar)
+
+        self.retranslateUi(MainWindow)
+
+        QMetaObject.connectSlotsByName(MainWindow)
+    # setupUi
+
+    def retranslateUi(self, MainWindow):
+        MainWindow.setWindowTitle(QCoreApplication.translate("MainWindow", u"MainWindow", None))
+        self.pushButton.setText(QCoreApplication.translate("MainWindow", u"转换", None))
+        self.pushButton_2.setText(QCoreApplication.translate("MainWindow", u"选择图片路径", None))
+        self.comboBox.setItemText(0, "")
+        self.comboBox.setItemText(1, QCoreApplication.translate("MainWindow", u"呐喊", None))
+        self.comboBox.setItemText(2, QCoreApplication.translate("MainWindow", u"星空", None))
+        self.comboBox.setItemText(3, QCoreApplication.translate("MainWindow", u"镶嵌风", None))
+        self.comboBox.setItemText(4, QCoreApplication.translate("MainWindow", u"构图VII", None))
+        self.comboBox.setItemText(5, QCoreApplication.translate("MainWindow", u"The muse", None))
+        self.comboBox.setItemText(6, QCoreApplication.translate("MainWindow", u"udnie", None))
+        self.comboBox.setItemText(7, QCoreApplication.translate("MainWindow", u"神奈川冲浪里", None))
+
+        self.label.setText(QCoreApplication.translate("MainWindow", u"请选择图片风格", None))
+        self.pushButton_3.setText(QCoreApplication.translate("MainWindow", u"保存转换后图片", None))
+    # retranslateUi
+
+class MainWindow(QMainWindow, Ui_MainWindow):
+    def __init__(self):
+        super(MainWindow, self).__init__()
+        self.setupUi(self)
+        self.image = None
+        self.transformed_image = None
+
+        #转换
+        self.pushButton.clicked.connect(self.transform_image)
+        self.pushButton_2.clicked.connect(self.load_image)
+        self.pushButton_3.clicked.connect(self.save_image)
+
+    def load_image(self):
+        options = QFileDialog.Options()
+        options |= QFileDialog.ReadOnly
+        file_name, _ = QFileDialog.getOpenFileName(self, "选择图片", "", "Images (*.png *.xpm *.jpg);;All Files (*)", options=options)
+        if file_name:
+            self.image = cv2.imread(file_name)
+            self.display_image(self.image, self.graphicsView)
+
+    def display_image(self, img, view):
+        img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+        h, w = view.height(), view.width()
+        img_resized = cv2.resize(img_rgb, (w, h))
+        h, w, ch = img_resized.shape
+        bytes_per_line = ch * w
+        q_img = QImage(img_resized.data, w, h, bytes_per_line, QImage.Format_RGB888)
+        pixmap = QPixmap.fromImage(q_img)
+        scene = QGraphicsScene()
+        scene.addItem(QGraphicsPixmapItem(pixmap))
+        view.setScene(scene)
+
+    def transform_image(self):
+        # 检查图像是否已加载
+        if self.image is not None:
+            #获取用户选择的风格
+            style = self.comboBox.currentIndex()
+            #使用OpenCV的深度学习模块cv2.dnn来加载预训练的神经网络模型
+            if style == 1:
+                #net = cv2.dnn.readNetFromTorch('../models/the_scream.t7') 因为exe文件在exe目录下，所以需要改一下路径，打包的时候使用的是这行。
+                net = cv2.dnn.readNetFromTorch('./models/the_scream.t7')
+            elif style == 2:
+                net = cv2.dnn.readNetFromTorch('./models/starry_night.t7')
+            elif style == 3:
+                net = cv2.dnn.readNetFromTorch('./models/mosaic.t7')
+            elif style == 4:
+                net = cv2.dnn.readNetFromTorch('./models/composition_vii.t7')
+            elif style == 5:
+                net = cv2.dnn.readNetFromTorch('./models/la_muse.t7')
+            elif style == 6:
+                net = cv2.dnn.readNetFromTorch('./models/udnie.t7')
+            elif style == 7:
+                net = cv2.dnn.readNetFromTorch('./models/the_wave.t7')
+            else:
+                QMessageBox.warning(self, "Warning", "请选择一个风格")
+                return
+
+            # 获取图像的高度和宽度
+            (h, w) = self.image.shape[:2]
+            # 将图像转换为blob，用于神经网络输入
+            blob = cv2.dnn.blobFromImage(self.image, 1.0, (w, h), (103.939, 116.779, 123.680), swapRB=False, crop=False)
+            net.setInput(blob)
+            out = net.forward() # 得到输出
+            out = out.reshape(3, out.shape[2], out.shape[3])
+            #恢复均值
+            out[0] += 103.939
+            out[1] += 116.779
+            out[2] += 123.68
+            #调整通道的顺序
+            out = out.transpose(1, 2, 0)
+            self.transformed_image = np.clip(out, 0, 255).astype('uint8')
+            self.display_image(self.transformed_image, self.graphicsView_2)
+
+    def save_image(self):
+        if self.transformed_image is not None:
+            options = QFileDialog.Options()
+            options |= QFileDialog.ShowDirsOnly
+            file_name, _ = QFileDialog.getSaveFileName(self, "保存图片", "", "Images (*.png *.xpm *.jpg);;All Files (*)", options=options)
+            if file_name:
+                cv2.imwrite(file_name, self.transformed_image)
+        else:
+            QMessageBox.warning(self, "Warning", "没有转换后的图片可保存")
+
+if __name__ == "__main__":
+    app = QApplication(sys.argv)
+    mainWin = MainWindow()
+    mainWin.show()
+    sys.exit(app.exec_())