|
|
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_())
|
