|
|
import sys
|
|
|
import warnings
|
|
|
import cv2
|
|
|
import numpy as np
|
|
|
import struct
|
|
|
import matplotlib.pyplot as plt
|
|
|
|
|
|
from PyQt5 import QtWidgets, QtGui
|
|
|
|
|
|
from ColorMenu import ColorMenu
|
|
|
from geometryMenu import geometryMenu
|
|
|
from mophologyMenu import mophologyMenu
|
|
|
from edgeMenu import EdgeMenu
|
|
|
from filterMenu import FilterMenu
|
|
|
from faceDetect import FaceDetect
|
|
|
from ui_file_1 import Ui_mainWindow
|
|
|
from PyQt5.QtWidgets import QMessageBox, QFileDialog, QInputDialog, QLineEdit
|
|
|
from qt_material import apply_stylesheet
|
|
|
|
|
|
|
|
|
class WXTWindow(QtWidgets.QMainWindow, Ui_mainWindow):
|
|
|
def __init__(self):
|
|
|
self.default_save_path = 'img'
|
|
|
super(WXTWindow, self).__init__()
|
|
|
self.setupUi(self)
|
|
|
#QMessageBox.about(self.window(), "使用提示", "欢迎使用~ 在对图像进行操作前，请先选择要处理的图片~")
|
|
|
self.pic_path = "" # 初始化空值
|
|
|
self.menubar.setEnabled(True)
|
|
|
self.color_menu.setEnabled(False)
|
|
|
self.geo_menu.setEnabled(False)
|
|
|
self.histogram_menu.setEnabled(False)
|
|
|
self.mophology_menu.setEnabled(False)
|
|
|
self.advancecd_menu.setEnabled(False)
|
|
|
self.denoising_menu.setEnabled(False)
|
|
|
self.faceMenu.setEnabled(False)
|
|
|
self.Save_btn.setEnabled(False)
|
|
|
|
|
|
# 打开图片
|
|
|
def OpenImg(self):
|
|
|
imgName, imgType = QFileDialog.getOpenFileName(self, "打开图片", "OriginPictures", "*.jpg;;*.bmp;;*.png")
|
|
|
jpg = QtGui.QPixmap(imgName).scaled(self.ori_pic.width(), self.ori_pic.height(), 1 ,1)
|
|
|
self.ori_pic.setPixmap(jpg)
|
|
|
|
|
|
self.origin_pic_path.setText(imgName)
|
|
|
self.pic_path = imgName
|
|
|
self.fixed_pic.setText("")
|
|
|
self.fixed_pic.clear()
|
|
|
|
|
|
if self.pic_path == "":
|
|
|
self.color_menu.setEnabled(False)
|
|
|
self.geo_menu.setEnabled(False)
|
|
|
self.histogram_menu.setEnabled(False)
|
|
|
self.mophology_menu.setEnabled(False)
|
|
|
self.advancecd_menu.setEnabled(False)
|
|
|
self.denoising_menu.setEnabled(False)
|
|
|
self.faceMenu.setEnabled(False)
|
|
|
self.Save_btn.setEnabled(False)
|
|
|
return
|
|
|
|
|
|
#各部分初始化
|
|
|
self.Initialize_color()
|
|
|
self.Initialize_geo()
|
|
|
self.InitializeMopho()
|
|
|
self.InitializeEdge()
|
|
|
self.InitializeFilter()
|
|
|
|
|
|
self.color_menu.setEnabled(True)
|
|
|
self.geo_menu.setEnabled(True)
|
|
|
self.histogram_menu.setEnabled(True)
|
|
|
self.mophology_menu.setEnabled(True)
|
|
|
self.advancecd_menu.setEnabled(True)
|
|
|
self.denoising_menu.setEnabled(True)
|
|
|
self.faceMenu.setEnabled(True)
|
|
|
self.Save_btn.setEnabled(True)
|
|
|
|
|
|
def SaveImg(self):
|
|
|
if self.save_pic_path.text() == "默认保存在img文件夹":
|
|
|
QMessageBox.information(self, "提示", "尚未进行图片操作，不能保存")
|
|
|
return
|
|
|
|
|
|
if self.pic_path != "" :
|
|
|
imgURL = QFileDialog.getExistingDirectory(self, "保存图片", "SavedPictures")
|
|
|
savepic = cv2.imread(self.save_pic_path.text(), 1)
|
|
|
cv2.imwrite(imgURL + '/SavedPic.jpg', savepic)
|
|
|
|
|
|
|
|
|
# 显示保存的图片
|
|
|
def Show_Fixed_Pic(self, path):
|
|
|
jpg = QtGui.QPixmap(path).scaled(self.fixed_pic.width(), self.fixed_pic.height(), 1, 1)
|
|
|
self.fixed_pic.setPixmap(jpg)
|
|
|
self.save_pic_path.setText(path)
|
|
|
|
|
|
# 显示位图信息头
|
|
|
def Show_Pic_Info(self):
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
#以下为色彩菜单函数
|
|
|
def Initialize_color(self):
|
|
|
self.color = ColorMenu(self.pic_path)
|
|
|
|
|
|
def GreyPic(self):
|
|
|
greypic = cv2.imread(self.pic_path, 0)
|
|
|
cv2.imwrite('img/greyimg.jpg', greypic)
|
|
|
self.Show_Fixed_Pic('img/greyimg.jpg')
|
|
|
|
|
|
def ExtractBchannel(self):
|
|
|
self.color.Bchannel()
|
|
|
self.Show_Fixed_Pic('img/colorImgs/Cchannel.jpg')
|
|
|
|
|
|
def ExtractGchannel(self):
|
|
|
self.color.Gchannel()
|
|
|
self.Show_Fixed_Pic('img/colorImgs/Cchannel.jpg')
|
|
|
|
|
|
def ExtractRchannel(self):
|
|
|
self.color.Rchannel()
|
|
|
self.Show_Fixed_Pic('img/colorImgs/Cchannel.jpg')
|
|
|
|
|
|
def ExtractHchannel(self):
|
|
|
self.color.Hchannel()
|
|
|
self.Show_Fixed_Pic('img/colorImgs/Cchannel.jpg')
|
|
|
|
|
|
def ExtractSchannel(self):
|
|
|
self.color.Schannel()
|
|
|
self.Show_Fixed_Pic('img/colorImgs/Cchannel.jpg')
|
|
|
|
|
|
def ExtractVchannel(self):
|
|
|
self.color.Vchannel()
|
|
|
self.Show_Fixed_Pic('img/colorImgs/Cchannel.jpg')
|
|
|
|
|
|
|
|
|
|
|
|
#以下为移动菜单函数
|
|
|
|
|
|
def Initialize_geo(self):
|
|
|
self.geoOperation = geometryMenu(self.pic_path)
|
|
|
|
|
|
def MovePic(self):
|
|
|
numx, ok1 = QInputDialog.getInt(self, 'X轴移动值', '请输入向右移动的值', 0, -10000, 10000, 10)
|
|
|
numy, ok2 = QInputDialog.getInt(self, 'Y轴移动值', '请输入向下移动的值', 0, -10000, 10000, 10)
|
|
|
|
|
|
if ok1 and ok2:
|
|
|
#if numx != 0 or numy != 0:
|
|
|
self.geoOperation.ImageMove(numx, numy)
|
|
|
self.Show_Fixed_Pic('img/geometryImgs/moved_pic.jpg')
|
|
|
|
|
|
def ImageResizeTimes(self):
|
|
|
numx, ok1 = QInputDialog.getDouble(self, '横向放缩值', '请输入横向放缩值', 1, 0.01, 5, 1)
|
|
|
numy, ok2 = QInputDialog.getDouble(self, '纵向放缩值', '请输入纵向放缩值', 1, 0.01, 5, 1)
|
|
|
|
|
|
if ok1 and ok2:
|
|
|
if numx != 1 or numy != 1:
|
|
|
self.geoOperation.ImageResizeTimes(numx, numy)
|
|
|
self.Show_Fixed_Pic('img/geometryImgs/moved_pic.jpg')
|
|
|
|
|
|
def ImageResizePixel(self):
|
|
|
numx, ok1 = QInputDialog.getInt(self, '横向像素值', '请输入横向像素值', 8, 8, 20000, 1)
|
|
|
numy, ok2 = QInputDialog.getInt(self, '纵向像素值', '请输入纵向像素值', 8, 8, 20000, 1)
|
|
|
|
|
|
if ok1 and ok2:
|
|
|
self.geoOperation.ImageResizePixel(numx, numy)
|
|
|
self.Show_Fixed_Pic('img/geometryImgs/moved_pic.jpg')
|
|
|
|
|
|
def HorizontalFlip(self):
|
|
|
self.geoOperation.HorizentalFlip()
|
|
|
self.Show_Fixed_Pic('img/geometryImgs/moved_pic.jpg')
|
|
|
|
|
|
def VerticalFlip(self):
|
|
|
self.geoOperation.VerticalFlip()
|
|
|
self.Show_Fixed_Pic('img/geometryImgs/moved_pic.jpg')
|
|
|
|
|
|
def CrossFlip(self):
|
|
|
self.geoOperation.CrossFlip()
|
|
|
self.Show_Fixed_Pic('img/geometryImgs/moved_pic.jpg')
|
|
|
|
|
|
def FreeRotate(self):
|
|
|
angle, ok1 = QInputDialog.getInt(self, '角度', '请输入顺时针旋转角度', 0, -360, 360, 1)
|
|
|
size, ok2 = QInputDialog.getDouble(self, '放缩值', '请输入放缩值,默认为1无需更改', 1, 0.01, 5, 1)
|
|
|
|
|
|
if ok1 and ok2:
|
|
|
self.geoOperation.FreeRotate(angle, size)
|
|
|
self.Show_Fixed_Pic('img/geometryImgs/moved_pic.jpg')
|
|
|
|
|
|
def FixedRotate(self):
|
|
|
items = ('90', '180', '270')
|
|
|
angle, ok = QInputDialog.getItem(self, '角度', '请输入顺时针旋转角度', items, 0, False)
|
|
|
|
|
|
if ok:
|
|
|
self.geoOperation.FixedRotate(angle)
|
|
|
self.Show_Fixed_Pic('img/geometryImgs/moved_pic.jpg')
|
|
|
|
|
|
|
|
|
|
|
|
#以下为直方图函数
|
|
|
def DrawGreyHistogram(self):
|
|
|
greypic = cv2.imread(self.pic_path, 0)
|
|
|
hist = cv2.calcHist([greypic], [0], None, [256], [0, 256])
|
|
|
plt.plot(hist)
|
|
|
plt.xlim([0, 255])
|
|
|
plt.savefig('img/histogram/histogram.jpg')
|
|
|
plt.close()
|
|
|
self.Show_Fixed_Pic('img/histogram/histogram.jpg')
|
|
|
|
|
|
def DrawRGBHistogram(self):
|
|
|
pic = cv2.imread(self.pic_path, 1)
|
|
|
color = ('r', 'g', 'b')
|
|
|
for i, col in enumerate(color):
|
|
|
hist = cv2.calcHist([pic], [i], None, [256], [0, 256])
|
|
|
plt.plot(hist)
|
|
|
plt.xlim([0, 256])
|
|
|
|
|
|
plt.savefig('img/histogram/histogram.jpg')
|
|
|
plt.close()
|
|
|
self.Show_Fixed_Pic('img/histogram/histogram.jpg')
|
|
|
|
|
|
|
|
|
|
|
|
#以下为形态学运算函数
|
|
|
def InitializeMopho(self):
|
|
|
self.mophology = mophologyMenu(self.pic_path)
|
|
|
|
|
|
def Erode(self):
|
|
|
size, ok = QInputDialog.getInt(self, '腐蚀结构元大小', '请输入，默认为5', 5, 1, 1000, 1)
|
|
|
|
|
|
if ok:
|
|
|
self.mophology.Erode(size)
|
|
|
self.Show_Fixed_Pic('img/mophologyImgs/mophoPic.jpg')
|
|
|
|
|
|
def Dilate(self):
|
|
|
size, ok = QInputDialog.getInt(self, '膨胀结构元大小', '请输入，默认为5', 5, 1, 1000, 1)
|
|
|
|
|
|
if ok:
|
|
|
self.mophology.Dilate(size)
|
|
|
self.Show_Fixed_Pic('img/mophologyImgs/mophoPic.jpg')
|
|
|
|
|
|
def OpenOperate(self):
|
|
|
size, ok = QInputDialog.getInt(self, '开运算结构元大小', '请输入，默认为5', 5, 1, 1000, 1)
|
|
|
|
|
|
if ok:
|
|
|
self.mophology.Open(size)
|
|
|
self.Show_Fixed_Pic('img/mophologyImgs/mophoPic.jpg')
|
|
|
def CloseOperate(self):
|
|
|
size, ok = QInputDialog.getInt(self, '闭运算结构元大小', '请输入，默认为5', 5, 1, 1000, 1)
|
|
|
|
|
|
if ok:
|
|
|
self.mophology.Close(size)
|
|
|
self.Show_Fixed_Pic('img/mophologyImgs/mophoPic.jpg')
|
|
|
|
|
|
|
|
|
|
|
|
#以下为图像增强以及边缘检测、锐化函数
|
|
|
def InitializeEdge(self):
|
|
|
self.edgeOpe = EdgeMenu(self.pic_path)
|
|
|
|
|
|
def PicStrength(self):
|
|
|
self.edgeOpe.PicStrength()
|
|
|
self.Show_Fixed_Pic('img/EdgeImgs/EdgePic.jpg')
|
|
|
|
|
|
def RobertsOperation(self):
|
|
|
self.edgeOpe.EdgeRoberts()
|
|
|
self.Show_Fixed_Pic('img/EdgeImgs/EdgePic.jpg')
|
|
|
|
|
|
def SobelOperation(self):
|
|
|
self.edgeOpe.EdgeSobel()
|
|
|
self.Show_Fixed_Pic('img/EdgeImgs/EdgePic.jpg')
|
|
|
|
|
|
def LaplacianOperation(self):
|
|
|
items = ('1', '3', '5', '7')
|
|
|
size, ok = QInputDialog.getItem(self, 'Laplacian算子大小', '请输入，默认为1', items, 0, False)
|
|
|
|
|
|
if ok:
|
|
|
self.edgeOpe.EdgeLaplacian(size)
|
|
|
self.Show_Fixed_Pic('img/EdgeImgs/EdgePic.jpg')
|
|
|
|
|
|
def LoGOperation(self):
|
|
|
QMessageBox.information(self, "提示", "LoG算子运算较慢请耐心等待")
|
|
|
|
|
|
self.edgeOpe.EdgeLoG()
|
|
|
self.Show_Fixed_Pic('img/EdgeImgs/EdgePic.jpg')
|
|
|
|
|
|
def CannyOperation(self):
|
|
|
self.edgeOpe.EdgeCanny()
|
|
|
self.Show_Fixed_Pic('img/EdgeImgs/EdgePic.jpg')
|
|
|
|
|
|
def HoughStraight(self):
|
|
|
colors = ['red', 'green', 'blue', 'black', 'white']
|
|
|
color, ok1 = QInputDialog.getItem(self, '线条颜色', '请输入，默认为红', colors, 0, False)
|
|
|
thres, ok2 = QInputDialog.getInt(self, '阈值', '请输入阈值，阈值越大检测出的线段越长越少，默认为100', 100, 1, 1000, 1)
|
|
|
|
|
|
if ok1 and ok2:
|
|
|
self.edgeOpe.HoughDetect(thres, color)
|
|
|
self.Show_Fixed_Pic('img/EdgeImgs/EdgePic.jpg')
|
|
|
|
|
|
def HoughCurve(self):
|
|
|
colors = ['red', 'green', 'blue', 'black', 'white']
|
|
|
color, ok1 = QInputDialog.getItem(self, '线条颜色', '请输入，默认为红', colors, 0, False)
|
|
|
thres, ok2 = QInputDialog.getInt(self, '阈值', '请输入阈值，阈值越大检测出的线段越长越少，默认为100', 100, 1, 1000, 1)
|
|
|
minline, ok3 = QInputDialog.getInt(self, '线段最小长度', '请输入线段最小长度，默认为0', 0, 0, 1000, 1)
|
|
|
maxgap, ok4 = QInputDialog.getInt(self, '线段间最大允许间隔', '请输入间隔默认为0', 0, 0, 1000, 1)
|
|
|
|
|
|
if ok1 and ok2 and ok3 and ok4:
|
|
|
self.edgeOpe.HoughDetectP(thres, minline, maxgap, color)
|
|
|
self.Show_Fixed_Pic('img/EdgeImgs/EdgePic.jpg')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#以下为去噪相关部分槽函数
|
|
|
def InitializeFilter(self):
|
|
|
self.filterOpe = FilterMenu(self.pic_path)
|
|
|
|
|
|
def GaussNoise(self):
|
|
|
avg, ok1 = QInputDialog.getDouble(self, '均值', '请输入均值', 0, 0, 100)
|
|
|
var, ok2 = QInputDialog.getDouble(self, '方差', '请输入方差', 0.5, 0, 100)
|
|
|
|
|
|
if ok1 and ok2:
|
|
|
self.filterOpe.GaussNoise(avg,var)
|
|
|
self.Show_Fixed_Pic('img/filterImgs/FilterPic.jpg')
|
|
|
|
|
|
def PepperSaltyNoise(self):
|
|
|
thres, ok1 = QInputDialog.getDouble(self, '噪声阈值', '请输入阈值', 0.2, 0.01, 0.99)
|
|
|
|
|
|
if ok1:
|
|
|
self.filterOpe.PepperSaltyNoise(thres)
|
|
|
self.Show_Fixed_Pic('img/filterImgs/FilterPic.jpg')
|
|
|
|
|
|
def GaussFilter(self):
|
|
|
sizes, ok = QInputDialog.getInt(self, '高斯核大小', '请输入,必须为奇数,默认为1', 1, 1, 15)
|
|
|
|
|
|
if ok:
|
|
|
if sizes % 2 == 1:
|
|
|
self.filterOpe.gaussFilter(sizes)
|
|
|
self.Show_Fixed_Pic('img/filterImgs/FilterPic.jpg')
|
|
|
else:
|
|
|
QMessageBox.information(self, "提示", "输入的核大小必须为奇数")
|
|
|
|
|
|
def AvgFilter(self):
|
|
|
self.filterOpe.AverageFilter()
|
|
|
self.Show_Fixed_Pic('img/filterImgs/FilterPic.jpg')
|
|
|
|
|
|
def MidFilter(self):
|
|
|
self.filterOpe.middleFilter()
|
|
|
self.Show_Fixed_Pic('img/filterImgs/FilterPic.jpg')
|
|
|
|
|
|
def minFilter(self):
|
|
|
self.filterOpe.LeastFilter()
|
|
|
self.Show_Fixed_Pic('img/filterImgs/FilterPic.jpg')
|
|
|
|
|
|
def maxFilter(self):
|
|
|
self.filterOpe.LargestFilter()
|
|
|
self.Show_Fixed_Pic('img/filterImgs/FilterPic.jpg')
|
|
|
|
|
|
def PassFilter(self):
|
|
|
items = ('0', '255')
|
|
|
|
|
|
min, ok1 = QInputDialog.getInt(self, '带通范围', '请输入最低范围,至少0', 20, 0, 255)
|
|
|
max, ok2 = QInputDialog.getInt(self, '带通范围', '请输入最高范围,至多255', 220, 0, 255)
|
|
|
num, ok3 = QInputDialog.getItem(self, '非滤指定值', '请选择', items, 0, False)
|
|
|
|
|
|
if ok1 and ok2 and ok3:
|
|
|
self.filterOpe.PassFilter(min, max, num)
|
|
|
self.Show_Fixed_Pic('img/filterImgs/FilterPic.jpg')
|
|
|
|
|
|
def BlockFilter(self):
|
|
|
items = ('0', '255')
|
|
|
|
|
|
min, ok1 = QInputDialog.getInt(self, '带阻范围', '请输入最低范围,至少0', 20, 0, 255)
|
|
|
max, ok2 = QInputDialog.getInt(self, '带阻范围', '请输入最高范围,至多255', 220, 0, 255)
|
|
|
num, ok3 = QInputDialog.getItem(self, '非滤指定值', '请选择', items, 0, False)
|
|
|
|
|
|
if ok1 and ok2 and ok3:
|
|
|
self.filterOpe.BlockFilter(min, max, num)
|
|
|
self.Show_Fixed_Pic('img/filterImgs/FilterPic.jpg')
|
|
|
|
|
|
#以下是人脸识别
|
|
|
def faceDetect(self):
|
|
|
self.facedtct = FaceDetect(self.pic_path)
|
|
|
self.facedtct.detectFace()
|
|
|
self.Show_Fixed_Pic('img/faceImg/facePic.jpg')
|
|
|
|
|
|
def videoDetect(self):
|
|
|
self.facedtct = FaceDetect(self.pic_path)
|
|
|
try:
|
|
|
self.facedtct.videoDetect()
|
|
|
except Exception as e:
|
|
|
QMessageBox.information(self, "提示", "退出了人脸识别")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
warnings.filterwarnings('ignore')
|
|
|
app = QtWidgets.QApplication(sys.argv)
|
|
|
myWin = WXTWindow()
|
|
|
apply_stylesheet(app, theme='light_cyan.xml')
|
|
|
myWin.show()
|
|
|
sys.exit(app.exec_()) |
