|
|
import cv2
|
|
|
import numpy as np
|
|
|
import random
|
|
|
import tkinter
|
|
|
import tkinter as tk
|
|
|
import matplotlib.pyplot as plt
|
|
|
import albumentations as A
|
|
|
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
|
|
|
from tkinter.messagebox import showinfo, showwarning
|
|
|
from tkinter import filedialog
|
|
|
from PIL import Image,ImageTk
|
|
|
from tkinter.filedialog import askopenfilename, askopenfilenames, asksaveasfilename, askdirectory
|
|
|
|
|
|
|
|
|
# ----------------------------------------------------------------------
|
|
|
plt.rcParams['font.sans-serif']=['SimHei'] #用来正常显示中文
|
|
|
plt.rcParams['axes.unicode_minus']=False #用来正常显示负号
|
|
|
picSize=400
|
|
|
img_open = None
|
|
|
img_result = None
|
|
|
img_empty = cv2.imread('data/image/OIP-C.jpg',0)
|
|
|
|
|
|
|
|
|
# ----------------------------------------------------------------------
|
|
|
# 几何变化类
|
|
|
# 水平翻转
|
|
|
def filp_h():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
transform = A.HorizontalFlip(p=1)
|
|
|
|
|
|
img_result = transform(image=np.array(img_open))
|
|
|
img_result = img_result['image']
|
|
|
placePic2(img_result,'水平翻转后')
|
|
|
|
|
|
# 垂直翻转
|
|
|
def flip_v():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
transform = A.VerticalFlip(p=1)
|
|
|
|
|
|
img_result = transform(image=np.array(img_open))
|
|
|
img_result = img_result['image']
|
|
|
|
|
|
placePic2(img_result,'垂直翻转后')
|
|
|
|
|
|
# 缩放
|
|
|
def resize():
|
|
|
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
paraLists = [["水平方向缩放后大小", 224], ["垂直方向缩放后大小", 224]]
|
|
|
paraW = paraWindow(root, "请设置水平和垂直方向的缩放尺寸", paraLists)
|
|
|
|
|
|
h = int(paraW.paraLists[0][1])
|
|
|
w = int(paraW.paraLists[1][1])
|
|
|
|
|
|
transform = A.Resize(h,w)
|
|
|
img_result = transform(image=np.array(img_open))
|
|
|
img_result = img_result['image']
|
|
|
|
|
|
placePic2(img_result,'缩放后')
|
|
|
|
|
|
# 旋转（可选择旋转角度）
|
|
|
def roate():
|
|
|
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
paraLists = [['请设置旋转角度',30]]
|
|
|
paraW = paraWindow(root, "旋转图片", paraLists)
|
|
|
|
|
|
angle = int(paraW.paraLists[0][1])
|
|
|
|
|
|
transform = A.Rotate(angle)
|
|
|
img_result = transform(image=np.array(img_open),p=1)
|
|
|
img_result = img_result['image']
|
|
|
|
|
|
placePic2(img_result,'旋转后')
|
|
|
|
|
|
# 裁剪（裁剪方式：centercrop、randomcrop和cutout）
|
|
|
def Centercrop():
|
|
|
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
paraLists = [["水平方向裁剪后大小", 224], ["垂直方向裁剪后大小", 224]]
|
|
|
paraW = paraWindow(root, "请设置水平和垂直方向的裁剪后尺寸", paraLists)
|
|
|
|
|
|
h = int(paraW.paraLists[0][1])
|
|
|
w = int(paraW.paraLists[1][1])
|
|
|
|
|
|
transform = A.CenterCrop(h, w,p=1)
|
|
|
img_result = transform(image=np.array(img_open))
|
|
|
img_result = img_result['image']
|
|
|
|
|
|
placePic2(img_result,'中心裁剪后')
|
|
|
|
|
|
def Randomcrop():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
paraLists = [["水平方向缩放后大小", 224], ["垂直方向缩放后大小", 224]]
|
|
|
paraW = paraWindow(root, "请设置水平和垂直方向的缩放尺寸", paraLists)
|
|
|
|
|
|
h = int(paraW.paraLists[0][1])
|
|
|
w = int(paraW.paraLists[1][1])
|
|
|
|
|
|
transform = A.RandomCrop(h, w,p=1)
|
|
|
img_result = transform(image=np.array(img_open))
|
|
|
img_result = img_result['image']
|
|
|
|
|
|
placePic2(img_result,'随机裁剪后')
|
|
|
|
|
|
# 反向裁剪（随机选择区域）
|
|
|
def cutout():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
h,w,c = img_open.shape
|
|
|
# 后续是原图和mask相乘，所以初始化为0
|
|
|
# 不需要裁剪的区域就不变，裁剪的区域变成0
|
|
|
mask = np.ones_like(img_open)
|
|
|
print(mask)
|
|
|
x = np.random.randint(0,w)
|
|
|
y = np.random.randint(0,h)
|
|
|
|
|
|
paraLists = [['相对原图裁剪区域的大小', 0.2]]
|
|
|
paraW = paraWindow(root, "请设置具体参数", paraLists)
|
|
|
# p = int(paraW.paraLists[0][1])
|
|
|
s = float(paraW.paraLists[0][1])
|
|
|
|
|
|
w_cut = int(s*w)
|
|
|
h_cut = int(s*h)
|
|
|
|
|
|
x1 = max(x - w_cut // 2, 0) # 计算裁剪的左上角坐标
|
|
|
x2 = min(x1 + w_cut, w) # 计算裁剪的右下角坐标
|
|
|
y1 = max(y - h_cut // 2, 0)
|
|
|
y2 = min(y1 + h_cut, h)
|
|
|
mask[y1:y2, x1:x2, :] = 0 # 将裁剪区域内像素值置为 0
|
|
|
|
|
|
|
|
|
img_result = img_open * mask # 与原始图像按元素相乘，得到 Cutout 后的图像
|
|
|
print(img_result[y1:y2, x1:x2, :])
|
|
|
placePic2(img_result,text2='反向随机裁剪后')
|
|
|
|
|
|
# 对比度、饱和度、亮度、色调变化
|
|
|
def ColorJitter():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
paraLists = [["亮度", 0], ["色调", 0],['对比度',0],['饱和度',0]]
|
|
|
paraW = paraWindow(root, "请设置具体参数", paraLists)
|
|
|
|
|
|
|
|
|
brightness = float(paraW.paraLists[0][1]) # 亮度
|
|
|
hue = float(paraW.paraLists[1][1]) # 色调
|
|
|
contrast = float(paraW.paraLists[2][1]) # 对比度
|
|
|
saturation = float(paraW.paraLists[3][1]) # 饱和度
|
|
|
|
|
|
transform = A.ColorJitter(brightness=brightness,hue=hue,saturation=saturation,contrast=contrast,p=1)
|
|
|
img_result = transform(image=np.array(img_open))
|
|
|
img_result = img_result['image']
|
|
|
placePic2(img_result)
|
|
|
|
|
|
# 转换成灰度图
|
|
|
def ToGray():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
placePic2(img_gray,'灰度图')
|
|
|
|
|
|
# 通道分离
|
|
|
def split_channel():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
''' 现在 的img_open 是RGB的'''
|
|
|
R,G,B = cv2.split(img_open)
|
|
|
paraLists = [['分离通道', 0]]
|
|
|
paraW = paraWindow(root, "012分别代表RGB", paraLists)
|
|
|
dict = {'0':R,'1':G,'2':B}
|
|
|
img_result = dict[str(int(paraW.paraLists[0][1]))]
|
|
|
placePic2(img_result,'分离通道后')
|
|
|
|
|
|
# 随机模糊
|
|
|
def Blur():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
paraLists = [['模糊的次数',3]]
|
|
|
paraW = paraWindow(root, "请设置具体参数", paraLists)
|
|
|
count = int(paraW.paraLists[0][1])
|
|
|
transform = A.Blur(blur_limit=count,p=1)
|
|
|
img_result = transform(image=np.array(img_open))
|
|
|
img_result = img_result['image']
|
|
|
placePic2(img_result)
|
|
|
|
|
|
# 高斯滤波
|
|
|
def Gaussblur():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
paraLists = [['模糊的次数', 3]]
|
|
|
paraW = paraWindow(root, "请设置具体参数", paraLists)
|
|
|
count = int(paraW.paraLists[0][1])
|
|
|
transform = A.GaussianBlur(blur_limit=count, p=1)
|
|
|
img_result = transform(image=np.array(img_open))
|
|
|
img_result = img_result['image']
|
|
|
placePic2(img_result,text2='经过高斯模糊后')
|
|
|
|
|
|
# 中值滤波
|
|
|
def Medianblur():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
paraLists = [['模糊的次数', 3]]
|
|
|
paraW = paraWindow(root, "请设置具体参数", paraLists)
|
|
|
count = int(paraW.paraLists[0][1])
|
|
|
transform = A.MedianBlur(blur_limit=count, p=1)
|
|
|
img_result = transform(image=np.array(img_open))
|
|
|
img_result = img_result['image']
|
|
|
placePic2(img_result,text2='经过中值滤波后')
|
|
|
|
|
|
# 添加高斯噪声
|
|
|
def add_gaussian_noise():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
paraLists = [['请输入均值', 0],['请输入标准差',30]]
|
|
|
paraW = paraWindow(root, "请设置具体参数", paraLists)
|
|
|
mean = int(paraW.paraLists[0][1])
|
|
|
sigma = int(paraW.paraLists[1][1])
|
|
|
|
|
|
# 生成高斯噪声矩阵
|
|
|
row, col, ch = img_open.shape
|
|
|
gaussian = np.random.randn(row, col, ch) * sigma + mean
|
|
|
gaussian = gaussian.reshape(row, col, ch)
|
|
|
img_result = img_open + gaussian
|
|
|
# 转换数据类型为8位无符号整数类型
|
|
|
img_result = cv2.convertScaleAbs(img_result)
|
|
|
placePic2(img_result,text2='添加高斯噪声后')
|
|
|
|
|
|
# 添加椒盐噪声
|
|
|
def add_salt_and_pepper_noise():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
paraLists = [['百分比', 0]]
|
|
|
paraW = paraWindow(root, "请设置具体参数", paraLists)
|
|
|
percentage = int(paraW.paraLists[0][1])
|
|
|
|
|
|
# 确保百分比在 0 到 100 之间
|
|
|
if percentage < 0 or percentage > 100:
|
|
|
showwarning(title='警告', message='百分比必须在 0 到 100 之间！')
|
|
|
return
|
|
|
|
|
|
# 生成椒盐噪声矩阵
|
|
|
row, col, ch = img_open.shape
|
|
|
noise = np.zeros((row, col, ch), np.uint8)
|
|
|
for i in range(row):
|
|
|
for j in range(col):
|
|
|
rand = np.random.randint(0, 100)
|
|
|
if rand < percentage:
|
|
|
noise[i][j] = [0, 0, 0]
|
|
|
elif rand > 100 - percentage:
|
|
|
noise[i][j] = [255, 255, 255]
|
|
|
else:
|
|
|
noise[i][j] = img_open[i][j]
|
|
|
|
|
|
# 将椒盐噪声矩阵添加到原始图像中
|
|
|
img_result = cv2.add(img_open, noise)
|
|
|
|
|
|
placePic2(img_result,text2='添加椒盐噪声后')
|
|
|
|
|
|
# 添加均值噪声
|
|
|
def add_mean_noise():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
paraLists = [['均值', 0], ['标准差', 30]]
|
|
|
paraW = paraWindow(root, "请设置具体参数", paraLists)
|
|
|
mean = int(paraW.paraLists[0][1])
|
|
|
std_dev = int(paraW.paraLists[1][1])
|
|
|
|
|
|
# 生成均值噪声矩阵
|
|
|
row, col, ch = img_open.shape
|
|
|
noise = np.random.normal(mean, std_dev, (row, col, ch)).astype(np.uint8)
|
|
|
|
|
|
# 将均值噪声矩阵添加到原始图像中
|
|
|
img_result = cv2.add(img_open, noise)
|
|
|
|
|
|
placePic2(img_result,text2='添加均值噪声后')
|
|
|
|
|
|
# 使用sobel算子进行锐化
|
|
|
def sobel():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
kernelx1 = np.array([[-1,-2,-1],[0,0,0],[1,2,1]], dtype=int)
|
|
|
kernely1 = np.array([[-1,0,1],[-2,0,2],[-1,0,1]], dtype=int)
|
|
|
x1 = cv2.filter2D(img_gray, cv2.CV_16S, kernelx1)
|
|
|
y1 = cv2.filter2D(img_gray, cv2.CV_16S, kernely1)
|
|
|
absX1 = cv2.convertScaleAbs(x1)
|
|
|
absY1 = cv2.convertScaleAbs(y1)
|
|
|
img_result = cv2.addWeighted(absX1, 0.5, absY1, 0.5, 0)
|
|
|
placePic2(img_result)
|
|
|
|
|
|
# 使用Prewitt算子进行锐化
|
|
|
def Prewitt():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
kernelx1 = np.array([[-1,0,1],[-1,0,1],[-1,0,1]], dtype=int)
|
|
|
kernely1 = np.array([[-1,-1,1],[0,0,0],[1,1,1]], dtype=int)
|
|
|
x1 = cv2.filter2D(img_gray, cv2.CV_16S, kernelx1)
|
|
|
y1 = cv2.filter2D(img_gray, cv2.CV_16S, kernely1)
|
|
|
absX1 = cv2.convertScaleAbs(x1)
|
|
|
absY1 = cv2.convertScaleAbs(y1)
|
|
|
img_result = cv2.addWeighted(absX1, 0.5, absY1, 0.5, 0)
|
|
|
placePic2(img_result)
|
|
|
|
|
|
# 使用robert算子进行锐化
|
|
|
def robert():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
kernelx1 = np.array([[-1, 0], [0, 1]], dtype=int)
|
|
|
kernely1 = np.array([[0, -1], [1, 0]], dtype=int)
|
|
|
x1 = cv2.filter2D(img_gray, cv2.CV_16S, kernelx1)
|
|
|
y1 = cv2.filter2D(img_gray, cv2.CV_16S, kernely1)
|
|
|
absX1 = cv2.convertScaleAbs(x1)
|
|
|
absY1 = cv2.convertScaleAbs(y1)
|
|
|
img_result = cv2.addWeighted(absX1, 0.5, absY1, 0.5, 0)
|
|
|
placePic2(img_result)
|
|
|
|
|
|
# 直方图均衡化
|
|
|
def equalhist():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
# 说明是灰度图像
|
|
|
if len(img_open)==2:
|
|
|
img_result = cv2.equalizeHist(img_open)
|
|
|
# RGB图像分别对每个通道都均衡化 然后在 合并
|
|
|
else:
|
|
|
b,g,r = cv2.split(img_open)
|
|
|
b_eq = cv2.equalizeHist(b)
|
|
|
g_eq = cv2.equalizeHist(g)
|
|
|
r_eq = cv2.equalizeHist(r)
|
|
|
|
|
|
img_result = cv2.merge([b_eq,g_eq,r_eq])
|
|
|
placePic2(img_result)
|
|
|
|
|
|
# 叠加图片操作
|
|
|
def mixup():
|
|
|
global img_open, img_gray, img_result
|
|
|
|
|
|
if img_open is None:
|
|
|
showwarning(title='警告', message='请先在文件菜单下打开图片！')
|
|
|
return
|
|
|
|
|
|
img_show1 = img_open
|
|
|
# 读取另一张图片
|
|
|
path = askopenfilename(title='打开另一张图片')
|
|
|
|
|
|
if path!=0 and path!='':
|
|
|
img_show2 = cv_imread(path)
|
|
|
|
|
|
if img_show2 is not None:
|
|
|
placePic2(img_show2)
|
|
|
else:
|
|
|
placePic2(img_empty)
|
|
|
showwarning(title='警告',message='请打开第二张图片')
|
|
|
|
|
|
# 下面进行mixup操作
|
|
|
paraLists = [['请输入一张图片所占比列', 0.5]]
|
|
|
paraW = paraWindow(root, "请设置具体参数", paraLists)
|
|
|
|
|
|
# 判断两张图片的shape 是否相同 不相同则都取小的那边
|
|
|
if img_show1.shape[:-1]!=img_show2.shape[:-1]:
|
|
|
showwarning(title='警告',message='两张图片大小不一样，会自动做裁剪使大小相同')
|
|
|
if img_show1.shape[0] > img_show2.shape[0]:
|
|
|
h = img_show2.shape[0]
|
|
|
else:
|
|
|
h = img_show1.shape[0]
|
|
|
|
|
|
if img_show1.shape[1] > img_show2.shape[1]:
|
|
|
w = img_show2.shape[1]
|
|
|
else:
|
|
|
w = img_show1.shape[1]
|
|
|
img_show1 = cv2.resize(img_show1,(h,w))
|
|
|
img_show2 = cv2.resize(img_show2,(h,w))
|
|
|
|
|
|
# 判断图片的channel是否相同，如果不相同 在全部转换为 RGB的
|
|
|
if len(img_show1)==2:
|
|
|
img_show1 = cv2.cvtColor(img_show1,cv2.COLOR_GRAY2RGB)
|
|
|
if len(img_show2)==2:
|
|
|
img_show2 = cv2.cvtColor(img_show1,cv2.COLOR_GRAY2RGB)
|
|
|
alpha = float(paraW.paraLists[0][1])
|
|
|
print(alpha)
|
|
|
# 这个方法需要两张图片的大小相同 需要做个resize操作
|
|
|
img_result = cv2.addWeighted(img_show1,alpha,img_show2,1-alpha,0)
|
|
|
cv_show('after mixup',img_result)
|
|
|
|
|
|
|
|
|
# ----------------------------------------------------------------------
|
|
|
def cv_show(name,img):
|
|
|
|
|
|
# 打开文件后会有img_open,img_gray的信息，img_result指进行操作后的可视化结果
|
|
|
# 图片很大时需要进行resize
|
|
|
h,w = img.shape[:2]
|
|
|
if h>1000 or w>1000:
|
|
|
img = cv2.resize(img,dsize=None,fx=0.8,fy=0.8)
|
|
|
|
|
|
cv2.imshow(name,img)
|
|
|
cv2.waitKey(0)
|
|
|
cv2.destroyAllWindows()
|
|
|
|
|
|
def cv_imread(filename,colorMode=cv2.IMREAD_COLOR):
|
|
|
|
|
|
# 读取文件路径中带汉字的图片,替代cv2.imread(filename)
|
|
|
# :param filename: 需要读取的文件
|
|
|
# :param colorMode: 彩色模式
|
|
|
# :return: 读取的图片
|
|
|
|
|
|
return cv2.imdecode(np.fromfile(filename, dtype=np.uint8), colorMode)
|
|
|
|
|
|
def cv_imwrite(path, img):
|
|
|
|
|
|
# 解决路径中有汉字的图片保存问题
|
|
|
# :param path: 保存的路径
|
|
|
# :param img: 保存的图片
|
|
|
|
|
|
cv2.imencode('.jpg', img)[1].tofile(path)
|
|
|
|
|
|
def placePic1(img_show, text1="原始图像"):
|
|
|
|
|
|
# 将打开的初始图像放置在左边窗格
|
|
|
# :param img_show: 需要显示的图片
|
|
|
# :param text1: 图片说明
|
|
|
|
|
|
global picSize
|
|
|
if img_show is None:
|
|
|
showwarning(title='警告', message='未打开图片！')
|
|
|
return
|
|
|
img = img_show.copy()
|
|
|
if len(img.shape) > 2:
|
|
|
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
|
|
|
height, width = img.shape[:2]
|
|
|
scaling = max(width, height) / picSize
|
|
|
newH = int(height / scaling)
|
|
|
newW = int(width / scaling)
|
|
|
img0 = Image.fromarray(img) # 由OpenCV图片转换为PIL图片格式
|
|
|
img0 = img0.resize((newW, newH))
|
|
|
img0 = ImageTk.PhotoImage(img0)
|
|
|
myWindow.originalText.set(text1)
|
|
|
myWindow.label3.config(image=img0)
|
|
|
myWindow.label3.image = img0
|
|
|
myWindow.label3.place(relx=0.25, rely=0.40, width=picSize, height=picSize, anchor=tkinter.CENTER) # 设置绝对座标
|
|
|
|
|
|
def placePic2(img_show, text2="处理结果图", RGB=True):
|
|
|
|
|
|
# 将处理结果图像放置在右边窗格
|
|
|
# :param img_show: 需要显示的图像
|
|
|
# :param text2: 图像文字说明
|
|
|
# :param RGB: 是否转换成RGB,默认转换
|
|
|
|
|
|
global picSize, img_result
|
|
|
if img_show is None:
|
|
|
showwarning(title='警告', message='没有结果图片！')
|
|
|
return
|
|
|
img = img_show.copy()
|
|
|
if len(img.shape) > 2 and RGB:
|
|
|
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
|
|
|
height, width = img.shape[:2]
|
|
|
# img = PointProcessing.global_linear_transmation(img)
|
|
|
scaling = max(width, height) / picSize
|
|
|
newH = int(height / scaling)
|
|
|
newW = int(width / scaling)
|
|
|
img0 = Image.fromarray(img) # 由OpenCV图片转换为PIL图片格式
|
|
|
img0 = img0.resize((newW, newH))
|
|
|
img0 = ImageTk.PhotoImage(img0)
|
|
|
myWindow.resultText.set(text2)
|
|
|
myWindow.label4.config(image=img0)
|
|
|
myWindow.label4.image = img0 # 通过标签来显示图片，使用ImageTK把img转成tk可以接收的形式
|
|
|
myWindow.label4.place(relx=0.75, rely=0.40, width=picSize, height=picSize, anchor=tkinter.CENTER)
|
|
|
|
|
|
def Choosepic():
|
|
|
# 打开文件选择对话框，并读取图片文件，存放在img_open全局变量中
|
|
|
global picSize, path_, img_open, img_gray, img_empty
|
|
|
path_ = askopenfilename(title="打开需要处理的图片")
|
|
|
# path.set(path_)
|
|
|
if path_ != 0 and path_ != '':
|
|
|
img_open = cv_imread(path_) # 读取文件路径中有汉字的图片文件，读取的是RGB图像
|
|
|
img_gray = cv2.cvtColor(img_open,cv2.COLOR_BGR2GRAY)
|
|
|
if img_open is not None:
|
|
|
placePic1(img_open)
|
|
|
else:
|
|
|
placePic1(img_empty)
|
|
|
showwarning(title='警告', message='无图片！')
|
|
|
else:
|
|
|
img_open = None
|
|
|
img_gray = None
|
|
|
placePic1(img_empty)
|
|
|
showwarning(title='警告', message='未选择图片！')
|
|
|
|
|
|
def Savepic():
|
|
|
# 打开文件对话框，将img_result图片保存到指定文件
|
|
|
global img_result
|
|
|
if img_result is not None:
|
|
|
fname = asksaveasfilename(title='保存文件', defaultextension='default.jpg',filetypes=[("JPG", ".jpg")])
|
|
|
if fname:
|
|
|
cv_imwrite(str(fname), img_result)
|
|
|
showinfo(title='提示', message='图片已保存！')
|
|
|
else:
|
|
|
showwarning(title='警告', message='请先输入文件名！')
|
|
|
else:
|
|
|
showwarning(title='警告', message='请先处理图片！')
|
|
|
|
|
|
def initWindows():
|
|
|
# 将系统界面恢复到初始界面,并不清除结果图片
|
|
|
myWindow.setVisibleLeft()
|
|
|
myWindow.setVisibleRight0()
|
|
|
myWindow.hideFig0()
|
|
|
myWindow.hideFig1()
|
|
|
|
|
|
if img_result is not None:
|
|
|
placePic2(img_result)
|
|
|
|
|
|
class ImageApp:
|
|
|
global picSize
|
|
|
|
|
|
def __init__(self, master):
|
|
|
self.master = master
|
|
|
self.master.title("图像多功能增强可视化窗口")
|
|
|
|
|
|
# 界面布局
|
|
|
self.init_menu(self.master) # 初始化菜单
|
|
|
self.init_widget(self.master) # 初始化需要用的控件（两个画布）
|
|
|
self.setVisibleLeft() # 显示左窗格
|
|
|
self.setVisibleRight0() # 显示右窗格
|
|
|
|
|
|
self.hideFig0() # 隐藏左窗格的画布myCanvas0
|
|
|
self.hideFig1() # 隐藏右窗格的画布myCanvas1
|
|
|
|
|
|
def init_widget(self, master):
|
|
|
self.originalText = tkinter.StringVar()
|
|
|
self.label1 = tkinter.Label(master, textvariable=self.originalText, font=("song", 12))
|
|
|
self.originalText.set("原始图像")
|
|
|
|
|
|
self.label3 = tkinter.Label(master, bg='gray86') # 用于显示原始图像，背景颜色设置成灰色
|
|
|
|
|
|
self.resultText = tkinter.StringVar()
|
|
|
self.label2 = tkinter.Label(master, textvariable=self.resultText, font=("song", 12))
|
|
|
self.resultText.set("处理结果图")
|
|
|
|
|
|
self.label4 = tkinter.Label(master, bg='gray86') # 用于显示处理结果，背景颜色设置成灰色
|
|
|
|
|
|
self.explainText = tkinter.StringVar()
|
|
|
|
|
|
self.myFig0 = plt.figure(figsize=(4, 4), facecolor='#dddddd')
|
|
|
self.myCanvas0 = FigureCanvasTkAgg(self.myFig0, master)
|
|
|
# myfig1在下边窗格的画布中显示多张包含图片的注释说明
|
|
|
self.myFig1 = plt.figure(figsize=(9, 1.8), facecolor='lightgray')
|
|
|
# myfig2在下边窗格中显示说明与单张图片
|
|
|
self.myFig2 = plt.figure(figsize=(4.5, 1.8), facecolor='lightgray')
|
|
|
self.myCanvas1 = FigureCanvasTkAgg(self.myFig1, master)
|
|
|
self.myCanvas2 = FigureCanvasTkAgg(self.myFig2, master)
|
|
|
|
|
|
def init_menu(self,master):
|
|
|
|
|
|
# 创建菜单和其内容
|
|
|
menubar = tk.Menu(master) # 创建菜单对象
|
|
|
|
|
|
# 创建菜单内容 后面加command 可以创建子菜单 到时候看看需不需要
|
|
|
fmenu_file = tk.Menu(master) # 文件
|
|
|
fmenu_file.add_command(label='打开',command=Choosepic)
|
|
|
fmenu_file.add_command(label='保存处理结果', command=Savepic)
|
|
|
fmenu_file.add_separator() # 添加一条分隔线
|
|
|
fmenu_file.add_command(label='退出',command=quit)
|
|
|
|
|
|
# 几何变换
|
|
|
fmenu_geo = tk.Menu(master)
|
|
|
fmenu_geo.add_command(label='水平翻转', command=filp_h)
|
|
|
fmenu_geo.add_command(label='垂直翻转', command=flip_v)
|
|
|
fmenu_geo.add_command(label='旋转', command=roate)
|
|
|
fmenu_geo.add_command(label='缩放', command=resize)
|
|
|
|
|
|
# 给裁剪菜单再次创建子菜单
|
|
|
submenu_crop = tk.Menu(fmenu_geo)
|
|
|
fmenu_geo.add_cascade(label='裁剪',menu=submenu_crop)
|
|
|
submenu_crop.add_command(label='中心裁剪',command=Centercrop)
|
|
|
submenu_crop.add_command(label='随机裁剪', command=Randomcrop)
|
|
|
submenu_crop.add_command(label='反向随机裁剪', command=cutout)
|
|
|
|
|
|
# 颜色空间变换
|
|
|
fmenu_sp = tk.Menu(master)
|
|
|
fmenu_sp.add_command(label='对比度变换',command=ColorJitter)
|
|
|
fmenu_sp.add_command(label='亮度变换', command=ColorJitter)
|
|
|
fmenu_sp.add_command(label='饱和度变换', command=ColorJitter)
|
|
|
fmenu_sp.add_command(label='色调',command=ColorJitter)
|
|
|
fmenu_sp.add_command(label='通道分离', command=split_channel)
|
|
|
fmenu_sp.add_command(label='灰度图', command=ToGray)
|
|
|
fmenu_sp.add_command(label='直方图均衡化', command=equalhist)
|
|
|
|
|
|
# 像素点操作
|
|
|
fmenu_pix = tk.Menu(master) # 创建像素点操作的主菜单
|
|
|
submenu_blur = tk.Menu(fmenu_pix) # 模糊子菜单
|
|
|
fmenu_pix.add_cascade(label='模糊',menu=submenu_blur) # 把模糊子菜单加到模糊下面
|
|
|
# 给模糊子菜单加子菜单
|
|
|
submenu_blur.add_command(label='高斯滤波',command=Gaussblur)
|
|
|
submenu_blur.add_command(label='随机模糊', command=Blur)
|
|
|
submenu_blur.add_command(label='中值滤波', command=Medianblur)
|
|
|
|
|
|
submenu_sh = tk.Menu(fmenu_pix)
|
|
|
fmenu_pix.add_cascade(label='锐化',menu=submenu_sh)
|
|
|
# 给锐化加子菜单
|
|
|
submenu_sh.add_command(label='robort算子',command=robert)
|
|
|
submenu_sh.add_command(label='Prewitt算子',command=Prewitt)
|
|
|
submenu_sh.add_command(label='Sobel算子',command=sobel)
|
|
|
|
|
|
submenu_noise = tk.Menu(fmenu_pix)
|
|
|
fmenu_pix.add_cascade(label='添加噪声',menu=submenu_noise)
|
|
|
# 给添加噪声加子菜单
|
|
|
submenu_noise.add_command(label='高斯噪声',command=add_gaussian_noise)
|
|
|
submenu_noise.add_command(label='椒盐噪声',command=add_salt_and_pepper_noise)
|
|
|
submenu_noise.add_command(label='均值噪声',command=add_mean_noise)
|
|
|
|
|
|
# 基于多张图片
|
|
|
fmenu_mul = tk.Menu(master)
|
|
|
fmenu_mul.add_command(label='mixup',command=mixup)
|
|
|
|
|
|
# 将菜单内容 添加到menubar
|
|
|
menubar.add_cascade(label="文件", menu=fmenu_file)
|
|
|
menubar.add_cascade(label="几何变化", menu=fmenu_geo)
|
|
|
menubar.add_cascade(label="颜色空间变换", menu=fmenu_sp)
|
|
|
menubar.add_cascade(label="频域像素点操作", menu=fmenu_pix)
|
|
|
menubar.add_cascade(label='基于多张图片',menu=fmenu_mul)
|
|
|
|
|
|
# 整体显示
|
|
|
master.config(menu=menubar)
|
|
|
|
|
|
def setVisibleLeft(self):
|
|
|
# label1是文字，label3目前是空只是占了位置，且背景颜色是灰色，需要放图片
|
|
|
|
|
|
self.label1.place(relx=0.25, rely=0.04, anchor=tkinter.CENTER) # 设置相对座标
|
|
|
self.label3.place(relx=0.25, rely=0.40, width=picSize, height=picSize,anchor=tkinter.CENTER) # 设置绝对座标
|
|
|
|
|
|
def setVisibleRight0(self):
|
|
|
# 右边窗格重叠的显示之一（right0）
|
|
|
# right0一般算法结果显示窗格
|
|
|
# right1-用于显示车牌识别结果
|
|
|
# myFig0-使用mayplot显示运行结果图片
|
|
|
# label2 是文字，label4目前是空只是占了位置 且背景颜色是灰色，需要放图片
|
|
|
|
|
|
self.label2.place(relx=0.75, rely=0.04, anchor=tkinter.CENTER)
|
|
|
self.label4.place(relx=0.75, rely=0.40, width=picSize, height=picSize,anchor=tkinter.CENTER)
|
|
|
|
|
|
def setVisibleBottom(self):
|
|
|
|
|
|
# 下边窗格的重叠显示之一（label5)
|
|
|
# label5-仅显示文字说明
|
|
|
# myFig1-用matplot显示多个文字图片
|
|
|
# explain-左边是label5,右边是myFig2
|
|
|
|
|
|
self.label5.place(relx=0.50, rely=0.86, width=900, height=180,anchor=tkinter.CENTER)
|
|
|
self.hideFig1()
|
|
|
self.hideExplain()
|
|
|
|
|
|
def hideFig0(self):
|
|
|
self.myCanvas0.get_tk_widget().place_forget() # 隐藏画布self
|
|
|
|
|
|
def hideFig1(self):
|
|
|
self.myCanvas1.get_tk_widget().place_forget() # 隐藏画布self
|
|
|
|
|
|
def hideExplain(self):
|
|
|
self.label5.config(wraplength=850)
|
|
|
self.myCanvas2.get_tk_widget().place_forget() # 隐藏画布self（暂时隐藏，用的时候可以用place方法恢复）
|
|
|
|
|
|
# 有的功能还需要选择具体参数，这个时候需要弹出一个小窗口给参数，并获取这些参数值，给具体的变化
|
|
|
|
|
|
|
|
|
# ----------------------------------------------------------------------
|
|
|
class paraWindow(tkinter.Toplevel):
|
|
|
|
|
|
# 多行一列的参数设置窗口
|
|
|
def __init__(self, root, title = None, paraLists=[]):
|
|
|
# Constructor
|
|
|
self.root = root
|
|
|
self.paraLists=paraLists
|
|
|
self.names = locals() #动态组件
|
|
|
tkinter.Toplevel.__init__(self,root)
|
|
|
if title:
|
|
|
self.title(title)
|
|
|
# 创建对话框的主体内容
|
|
|
frame = tkinter.Frame(self)
|
|
|
# 调用init_widgets方法来初始化对话框界面
|
|
|
self.initial_focus = self.init_widgets(frame)
|
|
|
frame.pack(padx=5, pady=5)
|
|
|
# 根据modal选项设置是否为模式对话框
|
|
|
self.grab_set() #重要，必须是模式对话框
|
|
|
# 为"WM_DELETE_WINDOW"协议使用self.cancel_click事件处理方法
|
|
|
self.protocol("WM_DELETE_WINDOW", self.cancel_click)
|
|
|
# 根据父窗口来设置对话框的位置
|
|
|
self.geometry("+%d+%d" % (root.winfo_rootx() + 100, root.winfo_rooty() + 100))
|
|
|
# 让对话框获取焦点
|
|
|
self.initial_focus.focus_set()
|
|
|
self.wait_window(self)
|
|
|
|
|
|
def init_widgets(self,master):
|
|
|
# 创建自定义对话框的内容
|
|
|
nrow=0
|
|
|
for i in range(len(self.paraLists)):
|
|
|
str0 = self.paraLists[i][0]
|
|
|
nrow = i + 1
|
|
|
labelMessage = tkinter.Label(master, text=str0, font=("song", 12))
|
|
|
labelMessage.grid(row=nrow, column=0)
|
|
|
self.names['paraV' + str(i)] = tkinter.StringVar()
|
|
|
self.names['paraE' + str(i)] = tkinter.Entry(master, textvariable=self.names['paraV' + str(i)], width=20)
|
|
|
self.names['paraV' + str(i)].set(self.paraLists[i][1])
|
|
|
self.names['paraE' + str(i)].grid(row=nrow, column=1) # 控件按列排列
|
|
|
b1 = tkinter.Button(master, text='确定退出', command=self.setPara)
|
|
|
b1.grid(row=nrow + 1, column=1)
|
|
|
self.bind("<Return>", self.setPara)
|
|
|
self.bind("<Escape>", self.cancel_click)
|
|
|
return self.names['paraE0']
|
|
|
|
|
|
def cancel_click(self, event=None):
|
|
|
showwarning(title='警告', message='必须先设置参数')
|
|
|
self.initial_focus.focus_set()
|
|
|
|
|
|
def setPara(self):
|
|
|
# 通过对话框设置参数
|
|
|
for i in range(len(self.paraLists)):
|
|
|
text0 = self.names['paraV' + str(i)].get()
|
|
|
if not self.on_validate(text0): # 如果不能通过校验，让用户重新输入
|
|
|
showwarning(title='警告', message='必须输入数字')
|
|
|
self.names['paraV' + str(i)].set(self.paraLists[i][1])
|
|
|
self.names['paraE'+ str(i)].focus_set()
|
|
|
return
|
|
|
else:
|
|
|
self.paraLists[i][1] = float(text0)
|
|
|
self.hide()
|
|
|
|
|
|
def on_validate(self,content):
|
|
|
# 该方法可对用户输入的数据进行校验，保证输入的是数字
|
|
|
for i in range(len(content) - 1, -1, -1):
|
|
|
if not (48 <= ord(content[i]) <= 57 or content[i] == "." or content[i] =="+" or content[i] =="-" ):
|
|
|
return False
|
|
|
return True
|
|
|
|
|
|
def hide(self):
|
|
|
# 销毁对话框
|
|
|
self.withdraw()
|
|
|
self.update_idletasks()
|
|
|
# 将焦点返回给父窗口
|
|
|
self.root.focus_set()
|
|
|
# 销毁自己
|
|
|
self.destroy()
|
|
|
self.root.update()
|
|
|
self.root.deiconify()
|
|
|
self.root.focus_set()
|
|
|
|
|
|
class paraWindow2(tkinter.Toplevel):
|
|
|
# 多行多列参数设置窗口
|
|
|
def __init__(self, root, paraLists,title,explain):
|
|
|
# Constructor
|
|
|
self.root = root
|
|
|
self.names=locals()
|
|
|
self.paraLists=paraLists
|
|
|
self.explain=explain
|
|
|
tkinter.Toplevel.__init__(self,root)
|
|
|
# self.geometry("400x300")
|
|
|
self.title(title)
|
|
|
# 调用init_widgets方法来初始化对话框界面
|
|
|
self.initial_focus = self.init_widgets()
|
|
|
# 根据modal选项设置是否为模式对话框
|
|
|
self.grab_set()
|
|
|
# 为"WM_DELETE_WINDOW"协议使用self.cancel_click事件处理方法
|
|
|
self.protocol("WM_DELETE_WINDOW", self.cancel_click)
|
|
|
# 根据父窗口来设置对话框的位置
|
|
|
self.geometry("+%d+%d" % (root.winfo_rootx() + 100, root.winfo_rooty() + 100))
|
|
|
# print(self.initial_focus)
|
|
|
# 让对话框获取焦点
|
|
|
self.initial_focus.focus_set()
|
|
|
self.wait_window(self)
|
|
|
|
|
|
def init_widgets(self):
|
|
|
frame1 = tkinter.Frame(self)
|
|
|
explainLabel = tkinter.Label(frame1, text=self.explain, font=("song", 12))
|
|
|
explainLabel.pack(padx=50, pady=5)
|
|
|
frame1.pack(padx=5, pady=5)
|
|
|
frame2 = tkinter.Frame(self)
|
|
|
nrow=1
|
|
|
for i in range(len(self.paraLists)):
|
|
|
nrow = i + 3
|
|
|
for j in range(len(self.paraLists[i])):
|
|
|
self.names['ev' + str(i) + str(j)] = tkinter.StringVar()
|
|
|
self.names['e' + str(i) + str(j)] = tkinter.Entry(frame2, textvariable=self.names['ev' + str(i) + str(j)],width=10)
|
|
|
self.names['ev' + str(i) + str(j)].set(self.paraLists[i][j])
|
|
|
self.names['e' + str(i) + str(j)].grid(row=nrow, column=j)
|
|
|
b1 = tkinter.Button(frame2, text='确定退出', command=self.setPara)
|
|
|
b1.grid(row=nrow + 1, column=1)
|
|
|
frame2.pack(padx=5, pady=6)
|
|
|
return b1
|
|
|
|
|
|
def setPara(self):
|
|
|
# 通过对话框给参数赋值
|
|
|
for i in range(len(self.paraLists)):
|
|
|
for j in range(len(self.paraLists[i])):
|
|
|
text0 = self.names['ev' + str(i) + str(j)].get()
|
|
|
if not self.on_validate(text0): # 如果不能通过校验，让用户重新输入
|
|
|
showwarning(title='警告', message='必须输入数字')
|
|
|
self.names['ev' + str(i) + str(j)].set(self.paraLists[i][j])
|
|
|
self.names['e' + str(i) + str(j)].focus_set()
|
|
|
return
|
|
|
else:
|
|
|
self.paraLists[i][j] = float(text0)
|
|
|
self.hide()
|
|
|
|
|
|
def on_validate(self,content):
|
|
|
# 该方法可对用户输入的数据进行校验，保证输入的是数字
|
|
|
for i in range(len(content) - 1, -1, -1):
|
|
|
if not (48 <= ord(content[i]) <= 57 or content[i] == "." or content[i] == "+" or content[i] == "-"):
|
|
|
return False
|
|
|
return True
|
|
|
|
|
|
def cancel_click(self, event=None):
|
|
|
showwarning(title='警告', message='必须先设置参数')
|
|
|
self.initial_focus.focus_set()
|
|
|
|
|
|
def hide(self):
|
|
|
self.withdraw()
|
|
|
self.update_idletasks()
|
|
|
# 将焦点返回给父窗口
|
|
|
self.root.focus_set()
|
|
|
# 销毁自己
|
|
|
self.destroy()
|
|
|
self.root.update()
|
|
|
self.root.deiconify()
|
|
|
self.root.focus_set()
|
|
|
|
|
|
class listWindow(tkinter.Toplevel):
|
|
|
# 下拉列表窗口，用对参数设置
|
|
|
def __init__(self, root, title = None, paraLists=[]):
|
|
|
# Constructor
|
|
|
self.root = root
|
|
|
self.paraLists=paraLists
|
|
|
tkinter.Toplevel.__init__(self,root)
|
|
|
if title:
|
|
|
self.title(title)
|
|
|
# 创建对话框的主体内容
|
|
|
# 调用init_widgets方法来初始化对话框界面
|
|
|
self.initial_focus = self.init_widgets(self)
|
|
|
# 根据modal选项设置是否为模式对话框
|
|
|
self.grab_set()
|
|
|
# 为"WM_DELETE_WINDOW"协议使用self.cancel_click事件处理方法
|
|
|
self.protocol("WM_DELETE_WINDOW", self.cancel_click)
|
|
|
# 根据父窗口来设置对话框的位置
|
|
|
self.geometry("+%d+%d" % (root.winfo_rootx() + 100, root.winfo_rooty() + 100))
|
|
|
# print(self.initial_focus)
|
|
|
# 让对话框获取焦点
|
|
|
self.initial_focus.focus_set()
|
|
|
self.wait_window(self)
|
|
|
|
|
|
def init_widgets(self,master):
|
|
|
str0 = self.paraLists[0][0]
|
|
|
labelMessage = tkinter.Label(master, text=self.paraLists[0][0], font=("song", 12))
|
|
|
labelMessage.grid(row=0, column=0)
|
|
|
self.listBox = tkinter.Listbox(master,selectmode=tkinter.SINGLE)
|
|
|
self.listBox.grid(row=1, column=0)
|
|
|
lists=self.paraLists[0][1]
|
|
|
for item in lists:
|
|
|
self.listBox.insert(tkinter.END, item)
|
|
|
self.listBox.select_set(0)
|
|
|
b1 = tkinter.Button(master, text='确定退出', command=self.setPara)
|
|
|
b1.grid(row=2, column=0)
|
|
|
self.bind("<Return>", self.setPara)
|
|
|
self.bind("<Escape>", self.cancel_click)
|
|
|
return self.listBox
|
|
|
|
|
|
def cancel_click(self, event=None):
|
|
|
showwarning(title='警告', message='必须先设置参数')
|
|
|
self.initial_focus.focus_set()
|
|
|
|
|
|
def setPara(self):
|
|
|
self.returnValue=self.listBox.curselection()[0]
|
|
|
self.hide()
|
|
|
|
|
|
def hide(self):
|
|
|
self.withdraw()
|
|
|
self.update_idletasks()
|
|
|
# 将焦点返回给父窗口
|
|
|
self.root.focus_set()
|
|
|
# 销毁自己
|
|
|
self.destroy()
|
|
|
self.root.update()
|
|
|
self.root.deiconify()
|
|
|
self.root.focus_set()
|
|
|
|
|
|
|
|
|
# ----------------------------------------------------------------------
|
|
|
if __name__ == "__main__":
|
|
|
root = tk.Tk()
|
|
|
root.geometry("1000x640+150+5") # 界面大小,以相对屏幕的坐标
|
|
|
root.resizable(0, 0)
|
|
|
myWindow = ImageApp(root)
|
|
|
root.mainloop() |
