You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
403 lines
17 KiB
403 lines
17 KiB
import tkinter as tk
|
|
import tkinter.filedialog
|
|
from tkinter.ttk import *
|
|
import cv2
|
|
import numpy as np
|
|
from PIL import Image, ImageTk
|
|
|
|
import Basic
|
|
import advanced
|
|
|
|
root = tk.Tk()
|
|
root.title('Galaxy_Eyes')
|
|
img = cv2.imread('origin.jpg')
|
|
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
|
|
img2 = Image.fromarray(img)
|
|
photo = ImageTk.PhotoImage(img2)
|
|
imageLabel = tk.Label(root, image=photo)
|
|
imageLabel.pack(side=tk.LEFT)
|
|
logical_var = tk.IntVar()
|
|
arithmetic_var = tk.IntVar()
|
|
imgx = np.zeros((img.shape[1], img.shape[0]))
|
|
|
|
|
|
def img_change(img_t):
|
|
global img
|
|
global img2
|
|
global photo
|
|
global imageLabel
|
|
img = img_t
|
|
img2 = Image.fromarray(img)
|
|
photo = ImageTk.PhotoImage(img2)
|
|
imageLabel.config(image=photo)
|
|
|
|
|
|
def ope():
|
|
filename = tkinter.filedialog.askopenfilename()
|
|
if filename != '':
|
|
i = cv2.imread(filename)
|
|
i = cv2.cvtColor(i, cv2.COLOR_BGR2RGB)
|
|
img_change(i)
|
|
|
|
|
|
def sav(imgt):
|
|
filename = tkinter.filedialog.asksaveasfilename(defaultextension='.png',
|
|
filetypes=(('PNG文件', '*.png'),
|
|
('JPEG文件', '*.jpg'),
|
|
('全部类型文件', '*.*')))
|
|
if filename != '':
|
|
img_t = imgt
|
|
if len(imgt.shape) >= 3:
|
|
img_t = cv2.cvtColor(img_t, cv2.COLOR_RGB2BGR)
|
|
cv2.imwrite(filename, img_t)
|
|
|
|
|
|
def otsu():
|
|
global img
|
|
img_change(Basic.turnToBinary(img))
|
|
|
|
|
|
def logical_operation(winNew, imx, imy):
|
|
if logical_var.get() == 1:
|
|
result = imx & imy
|
|
elif logical_var.get() == 2:
|
|
result = imx | imy
|
|
elif logical_var.get() == 3:
|
|
result = imx ^ imy
|
|
else:
|
|
return
|
|
result = cv2.resize(result, (600, 600))
|
|
img_change(result)
|
|
winNew.destroy()
|
|
|
|
|
|
def logical_image_choose(fr, ph):
|
|
filename = tkinter.filedialog.askopenfilename()
|
|
if filename != '':
|
|
imgt = cv2.imread(filename)
|
|
imgt = cv2.cvtColor(imgt, cv2.COLOR_BGR2RGB)
|
|
imgt = Basic.turnToBinary(imgt)
|
|
imgt = cv2.resize(imgt, (150, 150))
|
|
fr.photot = ImageTk.PhotoImage(Image.fromarray(imgt))
|
|
ph.config(image=fr.photot)
|
|
fr.imgy = imgt
|
|
|
|
|
|
def logical_window_setup():
|
|
global logical_var
|
|
logical_winNew = tk.Toplevel(root)
|
|
logical_winNew.wm_attributes('-topmost', 1)
|
|
logical_winNew.title('Logical_operation')
|
|
logical_winNew.geometry('400x400')
|
|
frame1 = tk.Frame(logical_winNew, relief=tkinter.RAISED, borderwidth=2)
|
|
operation_and = tk.Radiobutton(frame1, text='逻辑与', variable=logical_var, value=1)
|
|
operation_or = tk.Radiobutton(frame1, text='逻辑或', variable=logical_var, value=2)
|
|
operation_xor = tk.Radiobutton(frame1, text='逻辑异或', variable=logical_var, value=3)
|
|
operation_and.pack()
|
|
operation_or.pack()
|
|
operation_xor.pack()
|
|
frame1.pack()
|
|
frame2 = tk.Frame(logical_winNew, relief=tkinter.RAISED, borderwidth=2)
|
|
global img
|
|
img_Binary = Basic.turnToBinary(img)
|
|
img_Binary = cv2.resize(img_Binary, (150, 150)) # 缩小至100x100
|
|
frame2.imgx = img_Binary
|
|
frame2.imgy = img_Binary
|
|
frame2.photo1 = ImageTk.PhotoImage(Image.fromarray(img_Binary))
|
|
frame2.photo2 = frame2.photo1
|
|
photo1 = tk.Label(frame2, image=frame2.photo1)
|
|
photo2 = tk.Label(frame2, image=frame2.photo2)
|
|
photo1.pack(side=tk.LEFT)
|
|
photo2.pack(side=tk.RIGHT)
|
|
frame2.pack()
|
|
frame3 = tk.Frame(logical_winNew, relief=tkinter.RAISED, borderwidth=2)
|
|
choose_image = tk.Button(frame3, text='请选择操作图', command=lambda: logical_image_choose(frame2, photo2))
|
|
operation_button = tk.Button(frame3, text='执行操作',
|
|
command=lambda: logical_operation(logical_winNew, frame2.imgx, frame2.imgy))
|
|
choose_image.pack(side=tk.LEFT)
|
|
operation_button.pack(side=tk.RIGHT)
|
|
frame3.pack()
|
|
|
|
|
|
def arithmetic_operation(winNew, imx, imy):
|
|
if arithmetic_var.get() == 1:
|
|
result = cv2.add(imx, imy)
|
|
elif arithmetic_var.get() == 2:
|
|
result = cv2.subtract(imx, imy)
|
|
elif arithmetic_var.get() == 3:
|
|
result = cv2.multiply(imx, imy)
|
|
elif arithmetic_var.get() == 4:
|
|
result = cv2.divide(imx, imy)
|
|
else:
|
|
return
|
|
result = cv2.resize(result, (600, 600))
|
|
img_change(result)
|
|
winNew.destroy()
|
|
|
|
|
|
def arithmetic_image_choose(fr, ph):
|
|
filename = tkinter.filedialog.askopenfilename()
|
|
if filename != '':
|
|
imgt = cv2.imread(filename)
|
|
imgt = cv2.cvtColor(imgt, cv2.COLOR_BGR2RGB)
|
|
imgt = cv2.resize(imgt, (150, 150))
|
|
fr.photot = ImageTk.PhotoImage(Image.fromarray(imgt))
|
|
ph.config(image=fr.photot)
|
|
fr.imgy = imgt
|
|
|
|
|
|
def arithmetic_window_setup():
|
|
global arithmetic_var
|
|
arithmetic_winNew = tk.Toplevel(root)
|
|
arithmetic_winNew.wm_attributes('-topmost', 1)
|
|
arithmetic_winNew.title('Logical_operation')
|
|
arithmetic_winNew.geometry('400x400')
|
|
frame1 = tk.Frame(arithmetic_winNew, relief=tkinter.RAISED, borderwidth=2)
|
|
operation_add = tk.Radiobutton(frame1, text='算术加', variable=arithmetic_var, value=1)
|
|
operation_sub = tk.Radiobutton(frame1, text='算术减', variable=arithmetic_var, value=2)
|
|
operation_mul = tk.Radiobutton(frame1, text='算术乘', variable=arithmetic_var, value=3)
|
|
operation_div = tk.Radiobutton(frame1, text='算术除', variable=arithmetic_var, value=4)
|
|
operation_add.pack(side=tk.LEFT)
|
|
operation_sub.pack(side=tk.LEFT)
|
|
operation_mul.pack(side=tk.LEFT)
|
|
operation_div.pack(side=tk.LEFT)
|
|
frame1.pack()
|
|
frame2 = tk.Frame(arithmetic_winNew, relief=tkinter.RAISED, borderwidth=2)
|
|
global img
|
|
img = cv2.resize(img, (150, 150))
|
|
frame2.imgx = img
|
|
frame2.imgy = img
|
|
frame2.photo1 = ImageTk.PhotoImage(Image.fromarray(img))
|
|
frame2.photo2 = frame2.photo1
|
|
photo1 = tk.Label(frame2, image=frame2.photo1)
|
|
photo2 = tk.Label(frame2, image=frame2.photo2)
|
|
photo1.pack(side=tk.LEFT)
|
|
photo2.pack(side=tk.RIGHT)
|
|
frame2.pack()
|
|
frame3 = tk.Frame(arithmetic_winNew, relief=tkinter.RAISED, borderwidth=2)
|
|
choose_image = tk.Button(frame3, text='请选择操作图', command=lambda: arithmetic_image_choose(frame2, photo2))
|
|
operation_button = tk.Button(frame3, text='执行操作',
|
|
command=lambda: arithmetic_operation(arithmetic_winNew, frame2.imgx, frame2.imgy))
|
|
choose_image.pack(side=tk.LEFT)
|
|
operation_button.pack(side=tk.RIGHT)
|
|
frame3.pack()
|
|
|
|
|
|
def size_change_window_setup():
|
|
size_winNew = tk.Toplevel(root)
|
|
textlabel = tk.Label(size_winNew, text='选择放缩后高度和宽度的大小 和 使用的线性插值')
|
|
textlabel.pack()
|
|
frame1 = tk.Frame(size_winNew, relief=tkinter.RAISED, borderwidth=2)
|
|
frame1_hw = tk.Frame(frame1, relief=tkinter.RAISED, borderwidth=2) # 宽高
|
|
frame1_itp = tk.Frame(frame1, relief=tkinter.RAISED, borderwidth=2) # 线性插值
|
|
height_entry = tk.Entry(frame1_hw)
|
|
weight_entry = tk.Entry(frame1_hw)
|
|
global img
|
|
height_entry.insert(0, img.shape[0])
|
|
weight_entry.insert(0, img.shape[1])
|
|
frame1.pack()
|
|
frame1_hw.pack(side=tk.LEFT)
|
|
frame1_itp.pack(side=tk.LEFT)
|
|
height_entry.pack()
|
|
weight_entry.pack()
|
|
itp_var = tk.StringVar()
|
|
itp_comb = Combobox(size_winNew, textvariable=itp_var,
|
|
values=['最近邻插值', '双线性插值', '基于局部像素的重采样', '基于4x4像素邻域的3次插值法', '基于8x8像素邻域的Lanczos插值'])
|
|
itp_comb.pack()
|
|
tk.Button(size_winNew, text='放缩！',
|
|
command=lambda: img_change(
|
|
Basic.size_change(img, height_entry.get(), weight_entry.get(),
|
|
itp_comb.current()))).pack()
|
|
|
|
|
|
def panning_window_setup():
|
|
pan_winNew = tk.Toplevel(root)
|
|
framex = tk.Frame(pan_winNew, relief=tkinter.RAISED, borderwidth=2)
|
|
x_direction = tk.StringVar()
|
|
x_direction_comb = Combobox(framex, textvariable=x_direction, value=['上', '下'])
|
|
x_direction_comb.pack(side=tk.LEFT)
|
|
x_direction_comb.current(0)
|
|
x_direction_entry = tk.Entry(framex)
|
|
x_direction_entry.insert(0, 0)
|
|
x_direction_entry.pack(side=tk.LEFT)
|
|
framex.pack()
|
|
framey = tk.Frame(pan_winNew, relief=tkinter.RAISED, borderwidth=2)
|
|
y_direction = tk.StringVar()
|
|
y_direction_comb = Combobox(framey, textvariable=y_direction, value=['左', '右'])
|
|
y_direction_comb.pack(side=tk.LEFT)
|
|
y_direction_comb.current(0)
|
|
y_direction_entry = tk.Entry(framey)
|
|
y_direction_entry.pack(side=tk.LEFT)
|
|
y_direction_entry.insert(0, 0)
|
|
framey.pack()
|
|
global img
|
|
tk.Button(pan_winNew, text='平移！', command=lambda: img_change(
|
|
Basic.image_panning(img, int(x_direction_entry.get()) * (1 if x_direction.get() == '下' else -1),
|
|
int(y_direction_entry.get()) * (1 if y_direction.get() == '右' else -1)))).pack()
|
|
|
|
|
|
def rotate_window_setup():
|
|
rotate_winNew = tk.Toplevel(root)
|
|
rotate_direction = tk.StringVar()
|
|
rotate_direction_comb = Combobox(rotate_winNew, textvariable=rotate_direction, value=['逆时针', '顺时针'])
|
|
rotate_direction_comb.pack(side=tk.LEFT)
|
|
rotate_direction_comb.current(0)
|
|
rotate_angle = tk.Entry(rotate_winNew)
|
|
rotate_angle.insert(0, 0)
|
|
rotate_angle.pack(side=tk.LEFT)
|
|
global img
|
|
tk.Button(rotate_winNew, text='旋转！', command=lambda: img_change(
|
|
Basic.image_rotate(img, int(rotate_angle.get()) if rotate_direction.get() == '逆时针' else (
|
|
360 - int(rotate_angle.get()))))).pack(side=tk.BOTTOM)
|
|
|
|
|
|
def Matrix_operation(MA):
|
|
M = [[float for i in range(3)] for i in range(2)]
|
|
for i in range(0, 2):
|
|
for j in range(0, 3):
|
|
M[i][j] = float(MA[i][j].get())
|
|
M = np.float32(M)
|
|
global img
|
|
img_change(cv2.warpAffine(img, M, (img.shape[1], img.shape[0])))
|
|
|
|
|
|
def Matrix_window_setup():
|
|
Matrix_winNew = tk.Toplevel(root)
|
|
Matrix_A = [[tk.Entry() for i in range(3)] for i in range(2)]
|
|
for i in range(0, 2):
|
|
for j in range(0, 3):
|
|
Matrix_A[i][j] = tk.Entry(Matrix_winNew)
|
|
Matrix_A[i][j].grid(row=i, column=j)
|
|
Matrix_A[i][j].insert(0, (1 if i == j else 0))
|
|
for i in range(0, 3):
|
|
tk.Label(Matrix_winNew, text=(1 if i == 2 else 0)).grid(row=2, column=i)
|
|
tk.Button(Matrix_winNew, text='仿射变换', command=lambda: Matrix_operation(Matrix_A)).grid(row=3, column=0,
|
|
columnspan=3)
|
|
|
|
|
|
def flip_window_setup():
|
|
flip_winNew = tk.Toplevel(root)
|
|
flip_direction = tk.StringVar()
|
|
flip_direction_comb = Combobox(flip_winNew, textvariable=flip_direction, value=['水平翻转', '垂直翻转', '对角翻转'])
|
|
flip_direction_comb.current(0)
|
|
flip_direction_comb.pack()
|
|
flip_dic = {'水平翻转': 1, '垂直翻转': 0, '对角翻转': -1}
|
|
global img
|
|
tk.Button(flip_winNew, text='翻转！',
|
|
command=lambda: img_change(cv2.flip(img, flip_dic[flip_direction.get()], dst=None))).pack()
|
|
|
|
|
|
def img_sharp(x, winNew):
|
|
if x == 1:
|
|
img_change(Basic.gradient_img_sharpen(img))
|
|
elif x == 2:
|
|
img_change(Basic.Roberts_img_sharpen(img))
|
|
elif x == 3:
|
|
img_change(Basic.Prewitt_img_sharpen(img))
|
|
elif x == 4:
|
|
img_change(Basic.Sobel_img_sharpen(img))
|
|
elif x == 5:
|
|
img_change(Basic.Laplacian_img_sharpen(img))
|
|
elif x == 6:
|
|
img_change(Basic.LoG_img_sharpen(img))
|
|
elif x == 7:
|
|
img_change(Basic.Canny_img_sharpen(img))
|
|
winNew.destroy()
|
|
|
|
|
|
def sharp_window_setup():
|
|
sharp_winNew = tk.Toplevel(root)
|
|
sharp_way = tk.StringVar()
|
|
sharp_way_comb = Combobox(sharp_winNew, textvariable=sharp_way,
|
|
value=['梯度', 'Roberts算子', 'Prewitt算子', 'Sobel算子', 'Laplacian算子', 'LoG算子', 'Canny边缘检测'])
|
|
sharp_way_comb.current(0)
|
|
sharp_way_comb.pack()
|
|
sharp_dic = {'梯度': 1, 'Roberts算子': 2, 'Prewitt算子': 3, 'Sobel算子': 4, 'Laplacian算子': 5, 'LoG算子': 6, 'Canny边缘检测': 7}
|
|
tk.Button(sharp_winNew, text='检测！', command=lambda: img_sharp(sharp_dic[sharp_way.get()], sharp_winNew)).pack()
|
|
|
|
|
|
def maintain_face_detection(camera, winNew, panel, bk):
|
|
global imgx
|
|
success, imgx = camera.read() # 从摄像头读取照片
|
|
if success:
|
|
global img
|
|
imgx = cv2.cvtColor(imgx, cv2.COLOR_BGR2RGB) # 转换颜色从BGR到RGB
|
|
if bk.get() == 1:
|
|
imgx = advanced.face_detect_function(imgx)
|
|
elif bk.get() == 2:
|
|
imgx = advanced.removeBG(imgx, img)
|
|
current_image = Image.fromarray(imgx) # 将图像转换成Image对象
|
|
imgtk = ImageTk.PhotoImage(image=current_image)
|
|
panel.imgtk1 = imgtk
|
|
panel.config(image=imgtk)
|
|
winNew.after(1, lambda: maintain_face_detection(camera, winNew, panel, bk))
|
|
|
|
|
|
def real_time_face_detection():
|
|
global imgx
|
|
face_winNew = tk.Toplevel(root)
|
|
imageLabelt = tk.Label(face_winNew)
|
|
imageLabelt.pack()
|
|
frame1 = tk.Frame(face_winNew)
|
|
frame1.pack()
|
|
bk = tk.IntVar()
|
|
no_background = tk.Radiobutton(frame1, text='无背景,人脸检测', variable=bk, value=1)
|
|
with_background = tk.Radiobutton(frame1, text='以软件当前显示图片为背景', variable=bk, value=2)
|
|
no_background.pack(side=tk.LEFT)
|
|
with_background.pack(side=tk.LEFT)
|
|
tk.Button(face_winNew, text='拍照', command=lambda: sav(imgx)).pack()
|
|
camera = cv2.VideoCapture(0)
|
|
maintain_face_detection(camera, face_winNew, imageLabelt, bk)
|
|
face_winNew.mainloop()
|
|
camera.release()
|
|
|
|
|
|
mainmenu = tk.Menu(root) # 菜单栏
|
|
filemenu = tk.Menu(mainmenu) # 文件菜单
|
|
processmenu = tk.Menu(mainmenu) # 处理菜单
|
|
calculationmenu = tk.Menu(mainmenu) # 运算菜单
|
|
histmenu = tk.Menu(mainmenu) # 直方图
|
|
smoothmenu = tk.Menu(processmenu)
|
|
noisemenu = tk.Menu(processmenu)
|
|
structuringmenu = tk.Menu(mainmenu)
|
|
advancedmenu = tk.Menu(mainmenu)
|
|
mainmenu.add_cascade(label='文件', menu=filemenu)
|
|
mainmenu.add_cascade(label='基本处理', menu=processmenu)
|
|
mainmenu.add_cascade(label='运算', menu=calculationmenu)
|
|
mainmenu.add_cascade(label='直方图', menu=histmenu)
|
|
mainmenu.add_cascade(label='形态学操作', menu=structuringmenu)
|
|
mainmenu.add_cascade(label='进阶操作', menu=advancedmenu)
|
|
filemenu.add_command(label='打开', command=ope)
|
|
filemenu.add_command(label='保存', command=lambda: sav(img))
|
|
processmenu.add_command(label='自适应阈值法二值化', command=otsu)
|
|
processmenu.add_command(label='缩放', command=size_change_window_setup)
|
|
processmenu.add_command(label='平移', command=panning_window_setup)
|
|
processmenu.add_command(label='旋转', command=rotate_window_setup)
|
|
processmenu.add_command(label='翻转', command=flip_window_setup)
|
|
processmenu.add_command(label='平移和旋转（矩阵）——仿射变换', command=Matrix_window_setup)
|
|
processmenu.add_command(label='Hough变换检测直线', command=lambda: img_change(Basic.Hough_transform(img)))
|
|
processmenu.add_command(label='边缘检测', command=sharp_window_setup)
|
|
processmenu.add_cascade(label='平滑/滤噪', menu=smoothmenu)
|
|
processmenu.add_cascade(label='添加噪声', menu=noisemenu)
|
|
smoothmenu.add_command(label='空域平滑-邻域平均法', command=lambda: img_change(Basic.neighberhood_average_smooth(img)))
|
|
smoothmenu.add_command(label='空域平滑-中值滤波法', command=lambda: img_change(Basic.median_filtering_smooth(img)))
|
|
smoothmenu.add_command(label='频域平滑-理想低通滤波', command=lambda: img_change(Basic.Ideal_low_pass_filtering_smooth(img)))
|
|
smoothmenu.add_command(label='频域平滑-巴特沃斯低通滤波', command=lambda: img_change(Basic.ButterWorth_filtering_smooth(img)))
|
|
smoothmenu.add_command(label='频域平滑-高斯低通滤波', command=lambda: img_change(Basic.Gaussian_filtering_smooth(img)))
|
|
noisemenu.add_command(label='添加椒盐噪声', command=lambda: img_change(Basic.produce_pepper_salt_noise(img)))
|
|
noisemenu.add_command(label='添加高斯噪声', command=lambda: img_change(Basic.produce_Gaussian_noise(img)))
|
|
calculationmenu.add_command(label='逻辑运算', command=logical_window_setup)
|
|
calculationmenu.add_command(label='算术运算', command=arithmetic_window_setup)
|
|
histmenu.add_command(label='灰度直方图', command=lambda: img_change(Basic.get_gray_hist(img)))
|
|
histmenu.add_command(label='彩色直方图', command=lambda: img_change(Basic.get_colorful_hist(img)))
|
|
histmenu.add_command(label='分段线性变化', command=lambda: img_change(Basic.segmented_linear_transformation(img)))
|
|
structuringmenu.add_command(label='膨胀', command=lambda: img_change(Basic.image_erode(img)))
|
|
structuringmenu.add_command(label='腐蚀', command=lambda: img_change(Basic.image_dilate(img)))
|
|
structuringmenu.add_command(label='开运算', command=lambda: img_change(Basic.image_dilate(Basic.image_erode(img))))
|
|
structuringmenu.add_command(label='闭运算', command=lambda: img_change(Basic.image_erode(Basic.image_dilate(img))))
|
|
advancedmenu.add_command(label='人脸检测', command=lambda: img_change(advanced.face_detect_function(img)))
|
|
advancedmenu.add_command(label='实时更换背景/拍照', command=real_time_face_detection)
|
|
root.config(menu=mainmenu)
|
|
# root.bind('Button-3', popupmenu)
|
|
root.mainloop()
|