diff --git a/styleTransfer.py b/styleTransfer.py new file mode 100644 index 0000000..4902c5c --- /dev/null +++ b/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_())