pl5uwmvih
/
fu_dig
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ADD main.py file via upload

Browse Source
main
pl5uwmvih 5 months ago
parent 197fa1ca1e
commit 58d6cf2e85
1 changed files with 550 additions and 0 deletions
Show Stats Download Patch File Download Diff File

550
main.py
Unescape View File

@ -0,0 +1,550 @@
from PIL import Image as Im
from PIL import ImageTk
import tkinter as tk # 导入 Tkinter 库
from tkinter.filedialog import askopenfilename
import tkinter.messagebox
import tkinter.ttk
import tkinter.messagebox
from PIL import ImageDraw
from PIL import ImageFont
from PIL import ImageEnhance
from hig_class import *
from extra import wlw
import numpy as np
import tempfile
import os
class Win(tk.Tk):
def __init__(self):
super().__init__()
self.title('图像处理工具')
self.geometry('1080x720')
self.picture = None # self.picture将作为picture类的实例化对象
self.img = None # self.img将作为窗口类中一直变动的PIL对象图片
self.file_entry = None
self.setupUI()
def setupUI(self):
# 右边菜单栏
right_f = tk.Frame(self, height=720, width=360, bg="#F8F8FF")
right_f.pack(side=tk.RIGHT)
# 各种功能按钮名称及位置
btn1 = tk.Button(right_f, text='打开图像', bg="#BFEFFF", command=self.openToshow)
btn1.place(y=25, x=30, width=300, height=40)
btn2 = tk.Button(right_f, text='截图', bg="#BFEFFF", command=self.window_cut)
btn2.place(y=85, x=30, width=144, height=40)
btn3 = tk.Button(right_f, text='大小', bg="#BFEFFF", command=self.window_size)
btn3.place(y=85, x=186, width=144, height=40)
btn4 = tk.Button(right_f, text='旋转', bg='#B0E2FF', command=self.window_rotate)
btn4.place(y=145, x=30, width=144, height=40)
btn5 = tk.Button(right_f, text='翻转', bg='#B0E2FF', command=self.window_mirror)
btn5.place(y=145, x=186, width=144, height=40)
btn6 = tk.Button(right_f, text='灰度变换', bg='#B0E2FF', command=self.window_word)
btn6.place(y=205, x=30, width=144, height=40)
btn7 = tk.Button(right_f, text='滤镜', bg='#B0E2FF', command=self.window_style)
btn7.place(y=205, x=186, width=144, height=40)
btn8 = tk.Button(right_f, text='显示直方图', bg='#63B8FF', command=self.toshowhist)
btn8.place(y=265, x=30, width=144, height=40)
btn9 = tk.Button(right_f, text='图像相加', bg='#63B8FF', command=self.openToshow2)
btn9.place(y=265, x=186, width=144, height=40)
btn10 = tk.Button(right_f, text='滤波', bg='#63B8FF', command=self.openfilter)
btn10.place(y=325, x=30, width=144, height=40)
btn11 = tk.Button(right_f, text='Sobel锐化', bg='#63B8FF', command=self.opensharpen)
btn11.place(y=325, x=186, width=144, height=40)
# 各种调整名称及位置
lb1 = tk.Label(right_f, text='亮 度:', bg="#F8F8FF")
lb1.place(y=395, x=55, width=60, height=30)
self.inp1 = tk.Entry(right_f)
self.inp1.place(y=395, x=105, width=90, height=30)
lb2 = tk.Label(right_f, text='%', bg="#F8F8FF")
lb2.place(y=395, x=205, width=13, height=30)
b1 = tk.Button(right_f, text='确定', fg='#F8F8FF', width=8, command=self.brightnessPic, bg="#7EC0EE")
b1.place(y=395, x=240)
lb3 = tk.Label(right_f, text='色彩度:', bg="#F8F8FF")
lb3.place(y=440, x=55, width=60, height=30)
self.inp2 = tk.Entry(right_f)
self.inp2.place(y=440, x=105, width=90, height=30)
lb4 = tk.Label(right_f, text='%', bg="#F8F8FF")
lb4.place(y=440, x=205, width=13, height=30)
b2 = tk.Button(right_f, text='确定', fg='#F8F8FF', width=8, command=self.coolorPic, bg="#7EC0EE")
b2.place(y=444, x=240)
lb5 = tk.Label(right_f, text='对比度:', bg="#F8F8FF")
lb5.place(y=485, x=55, width=60, height=30)
self.inp3 = tk.Entry(right_f)
self.inp3.place(y=485, x=105, width=90, height=30)
lb6 = tk.Label(right_f, text='%', bg="#F8F8FF")
lb6.place(y=485, x=205, width=13, height=30)
b3 = tk.Button(right_f, text='确定', fg='#F8F8FF', width=8, command=self.contrastPic, bg="#7EC0EE")
b3.place(y=489, x=240)
lb7 = tk.Label(right_f, text='锐 度:', bg="#F8F8FF")
lb7.place(y=530, x=55, width=60, height=30)
self.inp4 = tk.Entry(right_f)
self.inp4.place(y=530, x=105, width=90, height=30)
lb8 = tk.Label(right_f, text='%', bg="#F8F8FF")
lb8.place(y=530, x=205, width=13, height=30)
b4 = tk.Button(right_f, text='确定', fg='#F8F8FF', width=8, command=self.sharpnessPic, bg="#7EC0EE")
b4.place(y=534, x=240)
# 底部恢复、保存、对比、人脸识别
btn18 = tk.Button(right_f, text='保存图像', fg='#F8F8FF', command=self.save_pic, bg="#1E90FF")
btn18.place(y=590, x=200, width=100, height=30)
btn19 = tk.Button(right_f, text='恢复图像', fg='#F8F8FF', command=self.replay, bg="#1E90FF")
btn19.place(y=590, x=30)
btn20 = tk.Button(right_f, text='对比图像', fg='#F8F8FF', command=self.compare, bg="#1E90FF")
btn20.place(y=590, x=120)
btn21 = tk.Button(right_f, text='人脸识别', fg='#F8F8FF', command=self.facedetec, bg="#1874CD")
btn21.place(y=640, x=30)
# 右侧图像显示栏
right_f = tk.Frame(self, height=720, width=720)
right_f.pack(side=tk.RIGHT)
self.image_l = tk.Label(right_f, relief='ridge')
self.image_l.place(x=0, y=0, width=720, height=720)
def openfilter(self):
filter_win = tk.Toplevel()
filter_win.title('滤波选择')
filter_win.geometry('150x150')
b1 = tk.Button(filter_win, text='均值滤波', command=self.filter1)
b1.place(y=30, x=35, width=75)
b2 = tk.Button(filter_win, text='中值滤波', command=self.filter2)
b2.place(y=60, x=35, width=75)
b3 = tk.Button(filter_win, text='完成', command=filter_win.destroy)
b3.place(y=110, x=80, width=40)
def opensharpen(self):
sharpen_pic = self.img
sharpen_pic = wlw.sharpen(sharpen_pic)
self.show_img(sharpen_pic)
# 打开图片时使用，传值（图）给展示函数
def openToshow(self):
address = self.getAddress()
self.open_picToimg = self.picture.open_pic(address)
self.firstPic(self.open_picToimg)
self.show_img(self.open_picToimg)
# 打开图片时使用，传值（图）给展示函数
def openToshow2(self):
address2 = self.getAddress2()
self.open_picToimg2 = self.picture.open_pic(address2)
self.firstPic2(self.open_picToimg2)
self.show_img2(self.open_picToimg2)
def filter1(self):
img_show = wlw.filter1(self.img)
self.image_l.config(image=img_show)
self.image_l.image = img_show
return self.img
def filter2(self):
img_show = wlw.filter2(self.img)
self.image_l.config(image=img_show)
self.image_l.image = img_show
return self.img
# 打开图片时使用，获得地址
def getAddress(self):
path = tk.StringVar()
self.file_entry = tk.Entry(self, state='readonly', text=path)
path_ = askopenfilename()
path.set(path_)
self.picture = picture()
return self.file_entry.get()
# 打开图片时使用，获得地址
def getAddress2(self):
path2 = tk.StringVar()
file_entry2 = tk.Entry(self, state='readonly', text=path2)
path_2 = askopenfilename()
path2.set(path_2)
self.picture2 = picture()
return file_entry2.get()
# 展示函数
def show_img(self, n_img):
self.img = n_img # self.img PIL对象方便传值给picture类以及本类中其他需要使用PIL图像的地方
img_show = ImageTk.PhotoImage(self.img)
self.image_l.config(image=img_show)
self.image_l.image = img_show
return self.img
def show_img2(self, n_img):
self.img2 = n_img # self.img PIL对象方便传值给picture类以及本类中其他需要使用PIL图像的地方
img_show2 = self.img2
img_show2 = wlw.Add(img_show2, self.Fpic)
self.image_l.config(image=img_show2)
self.image_l.image = img_show2
return self.img2
# 保存函数
def save_pic(self):
fname = tkinter.filedialog.asksaveasfilename(title='保存文件', filetypes=[("PNG", ".png")])
self.img.save(str(fname)) # PIL保存
# 打开图像调用
def open_pic(self, address):
self.pic_get = Im.open(address).convert('RGBA')
wid, hei = self.pic_get.size
if wid > 600 or hei > 400:
if tk.messagebox.askokcancel('提示', '图片可能过大，是否压缩？'):
needShow_pic = self.openResize()
return needShow_pic
return self.pic_get
else:
return self.pic_get
# 原图储存
def firstPic(self, pic):
self.Fpic = pic
return self.Fpic
# 原图储存
def firstPic2(self, pic):
self.Fpic2 = pic
return self.Fpic2
# 打开图像时的图像压缩展示
def openResize(self):
w, h = self.pic_get.size
w_hope = 500
h_hope = 300
f1 = 1.0 * w_hope / w
f2 = 1.0 * h_hope / h
factor = min([f1, f2])
width = int(w * factor)
height = int(h * factor)
pic_show = self.pic_get.resize((width, height))
return pic_show
# 截图操作页面
def window_cut(self):
Cut_win = tk.Toplevel()
Cut_win.title('截图操作')
Cut_win.geometry('220x380')
nowPic = self.img
if self.img == None:
lNone = tk.Label(Cut_win, text='请先打开一张图片')
lNone.place(y=55, x=50)
else:
wNow, hNow = nowPic.size
l1 = tk.Label(Cut_win, text="此时图片尺寸:")
l1.place(y=30, x=25)
l2 = tk.Label(Cut_win, text=wNow)
l2.place(y=55, x=25)
l3 = tk.Label(Cut_win, text="X")
l3.place(y=55, x=65)
l4 = tk.Label(Cut_win, text=hNow)
l4.place(y=55, x=85)
l5 = tk.Label(Cut_win, text="截图区域")
l5.place(y=85, x=25)
l6 = tk.Label(Cut_win, text="起始点：（左上为（0，0））")
l6.place(y=115, x=25)
l7 = tk.Label(Cut_win, text="x：")
l7.place(y=145, x=25)
self.e1 = tk.Entry(Cut_win, width=10)
self.e1.place(y=145, x=55)
l8 = tk.Label(Cut_win, text="y：")
l8.place(y=180, x=25)
self.e2 = tk.Entry(Cut_win, width=10)
self.e2.place(y=180, x=55)
l9 = tk.Label(Cut_win, text="终止点：")
l9.place(y=220, x=25)
l10 = tk.Label(Cut_win, text="x：")
l10.place(y=250, x=25)
self.e3 = tk.Entry(Cut_win, width=10)
self.e3.place(y=250, x=55)
l11 = tk.Label(Cut_win, text="y：")
l11.place(y=285, x=25)
self.e4 = tk.Entry(Cut_win, width=10)
self.e4.place(y=285, x=55)
b1 = tk.Button(Cut_win, text='确定', command=self.getCutpart)
b1.place(y=320, x=80, width=40)
b2 = tk.Button(Cut_win, text='完成', command=Cut_win.destroy)
b2.place(y=350, x=150, width=40)
# 大小尺寸操作窗口
def window_size(self):
Size_win = tk.Toplevel()
Size_win.title('尺寸操作')
Size_win.geometry('200x180')
l1 = tk.Label(Size_win, text="宽:")
l1.place(y=30, x=25)
self.text1 = tk.Entry(Size_win, width=10)
self.text1.place(y=25, x=50)
l1_1 = tk.Label(Size_win, text="px")
l1_1.place(y=28, x=150)
l2 = tk.Label(Size_win, text="高:")
l2.place(y=60, x=25)
self.text2 = tk.Entry(Size_win, width=10)
self.text2.place(y=55, x=50)
l2_1 = tk.Label(Size_win, text="px")
l2_1.place(y=58, x=150)
b1 = tk.Button(Size_win, text='确定', command=self.getSize_change)
b1.place(y=100, x=80, width=40)
b2 = tk.Button(Size_win, text='完成', command=Size_win.destroy)
b2.place(y=145, x=140, width=40)
# 接收截图区域,传送展示
def getCutpart(self):
nCut_pic = self.img
x = int(self.e1.get())
y = int(self.e2.get())
xl = int(self.e3.get())
yl = int(self.e4.get())
self.picture = picture()
showCut_pic = self.picture.Cutpic(nCut_pic, x, y, xl, yl)
self.show_img(showCut_pic)
# 获得输入尺寸
def getSize_change(self):
sizeNum_w = int(self.text1.get())
sizeNum_h = int(self.text2.get())
# print('1')
self.showSize_change(sizeNum_w, sizeNum_h)
# print(sizeNum_w, sizeNum_h)
# 尺寸修改并展示图片
def showSize_change(self, renewSize_w, renewSize_h):
# print('2')
self.picture = picture()
needResize_pic = self.img
show_resizePic = self.picture.changeResize(needResize_pic, renewSize_w, renewSize_h)
self.show_img(show_resizePic)
# 灰度转换功能窗口
def window_word(self):
Word_win = tk.Toplevel()
Word_win.title('灰度变化')
Word_win.geometry('300x400')
l1 = tk.Label(Word_win, text='请选择灰度转换方法：')
l1.place(y=10, x=20, width=130)
l0 = tk.Label(Word_win, text="n（n值）：")
l0.place(y=40, x=20)
self.i0 = tk.Entry(Word_win, width=13) # 输入参数n
self.i0.place(y=40, x=100)
bt1 = tk.Button(Word_win, text='n值化', command=self.getn)
bt1.place(y=70, x=20, width=80)
l2 = tk.Label(Word_win, text="k（对比度）：")
l2.place(y=110, x=20)
self.i1 = tk.Entry(Word_win, width=13) # 输入线性化参数k
self.i1.place(y=110, x=100)
l3 = tk.Label(Word_win, text="b（亮度）：")
l3.place(y=130, x=20)
self.i2 = tk.Entry(Word_win, width=13) # 输入线性化参数b
self.i2.place(y=130, x=100)
bt2 = tk.Button(Word_win, text='线性化', command=self.getkb)
bt2.place(y=160, x=20, width=80)
l3 = tk.Label(Word_win, text="c（尺度比较常数）：")
l3.place(y=200, x=20)
self.i3 = tk.Entry(Word_win, width=13) # 输入参数c
self.i3.place(y=200, x=140)
bt3 = tk.Button(Word_win, text='非线性化（对数转换)', command=self.getc)
bt3.place(y=230, x=20, width=120)
b7 = tk.Button(Word_win, text='完成', command=Word_win.destroy)
b7.place(y=300, x=210)
# 获得输入的n值转换值
def getn(self):
npic = self.img
n = float(self.i0.get())
# 图像线性化获取展示
npic = wlw.tonpic(npic, n)
self.show_img(npic)
# 获得输入的线性转换值
def getkb(self):
linearpic = self.img
k = float(self.i1.get())
b = float(self.i2.get())
# 图像线性化获取展示
linearpic = wlw.linearization(linearpic, k, b)
self.show_img(linearpic)
# 获得输入的对数转换值
def getc(self):
logpic = self.img
c = float(self.i3.get())
# 图像对数化获取展示
logpic = wlw.tologpic(logpic, c)
self.show_img(logpic)
# 图像旋转操作窗口
def window_rotate(self):
Rot_win = tk.Toplevel()
Rot_win.title('旋转')
Rot_win.geometry('225x220')
l1 = tk.Label(Rot_win, text="角度：")
l1.place(y=20, x=10)
self.inpt = tk.Entry(Rot_win, width=13)
self.inpt.place(y=15, x=50)
b0 = tk.Button(Rot_win, text='确定', command=lambda: self.getDegree('1'))
b0.place(y=55, x=160, width=40)
b1 = tk.Button(Rot_win, text='+90', command=lambda: self.getDegree('2'))
b1.place(y=85, x=60, width=95)
b2 = tk.Button(Rot_win, text='-90', command=lambda: self.getDegree('3'))
b2.place(y=115, x=60, width=95)
b3 = tk.Button(Rot_win, text='180', command=lambda: self.getDegree('4'))
b3.place(y=145, x=60, width=95)
b4 = tk.Button(Rot_win, text='完成', command=Rot_win.destroy)
b4.place(y=180, x=160, width=40)
# 旋转角度获取
def getDegree(self, n):
self.picture = picture()
needRotate_pic = self.img
# print('99')
# print(n)
if n == '1':
shouldDegree = float(self.inpt.get())
# print('36')
# print(shouldDegree)
showRotate_pic = self.picture.rotatePic(needRotate_pic, shouldDegree)
elif n == '2':
# print('34')
showRotate_pic = self.picture.rotatePic(needRotate_pic, +90)
elif n == '3':
showRotate_pic = self.picture.rotatePic(needRotate_pic, -90)
elif n == '4':
showRotate_pic = self.picture.rotatePic(needRotate_pic, 180)
else:
return 0
self.show_img(showRotate_pic)
# 滤镜选择窗口
def window_style(self):
Sty_win = tk.Toplevel()
Sty_win.title('滤镜选择')
Sty_win.geometry('230x180')
bt1 = tk.Button(Sty_win, text='图像模糊', command=self.sty_1)
bt1.place(y=25, x=25, width=80)
bt2 = tk.Button(Sty_win, text='轮廓滤波', command=self.sty_2)
bt2.place(y=25, x=115, width=80)
bt3 = tk.Button(Sty_win, text='高斯模糊', command=self.sty_3)
bt3.place(y=65, x=25, width=80)
bt4 = tk.Button(Sty_win, text='浮雕滤镜', command=self.sty_4)
bt4.place(y=65, x=115, width=80)
bt6 = tk.Button(Sty_win, text='完成', command=Sty_win.destroy)
bt6.place(y=140, x=160)
# 图像模糊获取展示
def sty_1(self):
sty_1_pic = self.img
relSty_1 = wlw.blurPic(sty_1_pic)
self.show_img(relSty_1)
# 边界增强获取展示
def sty_2(self):
sty_2_pic = self.img
reSty_2 = wlw.edge(sty_2_pic)
self.show_img(reSty_2)
# 高斯模糊获取展示
def sty_3(self):
sty_3_pic = self.img
reSty_3 = wlw.gaussianBlur(sty_3_pic)
self.show_img(reSty_3)
# 浮雕滤镜获取展示
def sty_4(self):
sty_4_pic = self.img
reSty_4 = wlw.emboss(sty_4_pic)
self.show_img(reSty_4)
# 亮度调整
def brightnessPic(self):
self.picture = picture()
needBright_pic = self.img
b_num = float(self.inp1.get())
briNum = b_num / 100
showBright_pic = self.picture.brightPic(needBright_pic, briNum)
self.show_img(showBright_pic)
# 色彩度调整
def coolorPic(self):
self.picture = picture()
needColor_pic = self.img
co_num = float(self.inp2.get())
colNum = co_num / 100
showColor_pic = self.picture.colornPic(needColor_pic, colNum)
self.show_img(showColor_pic)
# 对比度调整
def contrastPic(self):
self.picture = picture()
needCon_pic = self.img
c_num = float(self.inp3.get())
ConNum = c_num / 100
showContrast_pic = self.picture.constractPic(needCon_pic, ConNum)
self.show_img(showContrast_pic)
# 锐度调整
def sharpnessPic(self):
self.picture = picture()
needSharp_pic = self.img
s_num = float(self.inp4.get())
ShNum = s_num / 100
showSharp_pic = self.picture.constractPic(needSharp_pic, ShNum)
self.show_img(showSharp_pic)
# 镜像操作窗口
def window_mirror(self):
Mir_win = tk.Toplevel()
Mir_win.title('镜像操作')
Mir_win.geometry('150x150')
b1 = tk.Button(Mir_win, text='左右', command=self.MirrorImg_lr)
b1.place(y=30, x=35, width=75)
b2 = tk.Button(Mir_win, text='上下', command=self.MirrorImg_tb)
b2.place(y=60, x=35, width=75)
b3 = tk.Button(Mir_win, text='完成', command=Mir_win.destroy)
b3.place(y=110, x=80, width=40)
# 镜像左右调用展示
def MirrorImg_lr(self):
self.picture = picture()
Mirror_img_lr = self.img
MittotImg_lrFinish = self.picture.MirrorPic_leftOrright(Mirror_img_lr)
self.show_img(MittotImg_lrFinish)
# 镜像上下调用展示
def MirrorImg_tb(self):
self.picture = picture()
Mirror_img_tb = self.img
MittotImg_tbFinish = self.picture.MirrorPic_topOrbuttom(Mirror_img_tb)
self.show_img(MittotImg_tbFinish)
# 恢复图像
def replay(self):
self.show_img(self.Fpic)
# 对比图像
def compare(self):
Im._show(self.Fpic)
def toshowhist(self):
# gethist = wlw.showhist(self.img)
# self.show_img(gethist)
wlw.showhist(self.img)
#人脸识别
def facedetec(self):
minc=self.img
path=self.file_entry.get()
minc = wlw.face_detect(minc, path)
self.show_img(minc)
print("图片路径：", path)
if __name__ == '__main__':
root = Win()
# 窗体主循环
root.mainloop()
Write Preview
Loading…
Cancel
Save