from warnings import simplefilter simplefilter(action='ignore', category=FutureWarning) from tkinter import ttk import tkinter import cv2 import numpy as np from tkinter import * from PIL import Image,ImageTk from model_improved_alexnet import ImprovedAlexNet#调用模型 from tkinter.filedialog import askopenfilename import tkinter.messagebox import torch from predict_improved_alexnet import improved_alexnet_predict#调用预测函数 import warnings import os warnings.filterwarnings("ignore") from tkinter import * import tkinter.simpledialog import tkinter as tk import tkinter.simpledialog as sd from PIL import Image,ImageTk import matplotlib.pyplot as plt#画图库 plt.rcParams['font.sans-serif']=['PingFang'] #plt.rcParams['axes.unicode_minus'] = False class CustomDialog(sd.Dialog): def __init__(self, parent, title=None, background_image='bg.png', user_icon_path='user.gif', password_icon_path='hmenu-lock.gif'): self.background_image = background_image self.user_icon_path = user_icon_path self.password_icon_path = password_icon_path super().__init__(parent, title) def body(self, master): # 加载背景图 bg_image = Image.open(self.background_image) bg_photo = ImageTk.PhotoImage(bg_image) new_width = 400 new_height = 150 style = ttk.Style() style.configure('ExtraLarge.TButton', font=('Arial', 16), padding=15) self.bg_label = tk.Label(master, image=bg_photo) self.bg_label.image = bg_photo # 保持对图片的引用 self.bg_label.place(x=0, y=0, relwidth=10, relheight=10) # 背景标签覆盖整个对话框区域 # 调整对话框尺寸 self.geometry(f"{new_width}x{new_height}") # 加载用户名图标 user_icon_image = Image.open(self.user_icon_path) user_icon_photo = ImageTk.PhotoImage(user_icon_image) # 加载密码图标 password_icon_image = Image.open(self.password_icon_path) password_icon_photo = ImageTk.PhotoImage(password_icon_image) # 创建带用户名图标的标签 label_with_user_icon = tk.Label(master, text="用户名:", image=user_icon_photo, compound='left', font=('Arial', 16)) label_with_user_icon.grid(row=0) label_with_user_icon.image = user_icon_photo # 保持对用户名图标图片的引用 # 创建带密码图标的标签 label_with_password_icon = tk.Label(master, text="密码:", image=password_icon_photo, compound='left', font=('Arial', 16)) label_with_password_icon.grid(row=1) label_with_password_icon.image = password_icon_photo # 保持对密码图标图片的引用 #创建用于输入用户名和密码的文本框,字体为arial，大小为16 self.entry1 = tk.Entry(master, font=('Arial', 16)) self.entry2 = tk.Entry(master, font=('Arial', 16), show='*')#密码显示为* #确定文本框位置 self.entry1.grid(row=0, column=1) self.entry2.grid(row=1, column=1) return self.entry1 # 设置初始焦点 #获取用户输入的两个值并将它们作为一个元组存储在self.result变量中 def apply(self): first_value = self.entry1.get() second_value = self.entry2.get() self.result = (first_value, second_value) # 设置窗体 root = Tk()#创建主窗口 root.withdraw()#隐藏主窗口 dialog = CustomDialog(root, "登录")#显示登陆窗口 result = dialog.result #print("输入框1的值：", result[0]) #print("输入框2的值：", result[1]) if result is not None and str(result[0])=='12345' and str(result[1]) == '12345': tkinter.messagebox.showinfo('提示：', '欢迎进入 中药识别') root = Toplevel() else: tkinter.messagebox.showinfo('提示：', '登录失败') exit(-1) # 设置窗体 #root = Tk() root.title("") # 图片对话框信息 path = StringVar() file_entry = Entry(root, state='readonly', text=path) # 初始化 global now_img, alexnet_model alexnet_model = None #训练结果展示 def btn_pc(): image = plt.imread('./logs/0005.png') plt.figure()#创建图形窗口 plt.imshow(image)#将读取的图片显示在图形窗口上 plt.axis('off')#关闭了图形的坐标轴显示 plt.title('模型训练曲线01-70')#标题 image = plt.imread('./logs/0006.png') plt.figure() plt.imshow(image) plt.axis('off') plt.title('模型混淆矩阵01-70') image = plt.imread('./logs/0007.png') plt.figure() plt.imshow(image) plt.axis('off') plt.title('模型混淆矩阵71-100') image = plt.imread('./logs/0008.png') plt.figure() plt.imshow(image) plt.axis('off') plt.title('模型混淆矩阵71-100') plt.show() # 选择图片 def choosepic(): default_dir = r"./test/" path_ = askopenfilename(title=u'选择文件', initialdir=(os.path.expanduser(default_dir))) if len(path_) < 1:#若所选文件路径长度小于1，即代表没有这个文件 return path.set(path_)#路径赋值给path global now_img # 设置图片路径 now_img = file_entry.get() #print(now_img) img_open = Image.open(file_entry.get())#打开图片 img_open = img_open.resize((1100, 700))#调整大小 # 显示图片到窗体 img = ImageTk.PhotoImage(img_open)#将打开的图片转换为Tkinter可用的 image_label.config(image=img)#将图片显示在窗口中 image_label.image = img#将转换后的图片对象保存在image_label中，显示在窗口中 # 退出软件 def btn_exit(): exit(0) # 调用模型 def btn_ai(): global now_img global alexnet_model if alexnet_model is None: # create model alexnet_model = ImprovedAlexNet(num_classes=11) # load model weights model_weight_path = "./models/se-inc-alexnet.pth"#调用模型 alexnet_model.load_state_dict(torch.load(model_weight_path,map_location='cpu')) alexnet_model.eval() res=improved_alexnet_predict(now_img,alexnet_model)#调用预测函数进行并预测 s=str(res)#转化为字符形式 tkinter.messagebox.showinfo('提示：',s)#显示一个提示框，包含预测结果 # 显示识别率 def btn_acc(): res = '' with open('./logs/model.txt', 'r', encoding='utf-8') as fb: s = fb.read() res = res + '识别率是:%.2f%%'%(float(s)*100) tkinter.messagebox.showinfo('提示：', res) # 设置窗体 可调整 主窗口 mainframe = ttk.Frame(root, padding="5 4 12 12") mainframe.grid(column=0, row=0, sticky=(N, W, E, S)) mainframe.columnconfigure(0, weight=1) mainframe.rowconfigure(0, weight=1)#使得当窗口调整大小时，该列/行也会相应地调整大小 # 配置按钮，联系上述对应功能 style = ttk.Style() style.configure('ExtraLarge.TButton', font=('Arial', 16), padding=15) #创建按钮 ttk.Button(mainframe, text="模型训练", command=btn_pc, style='ExtraLarge.TButton').grid(column=1, row=4, sticky=W)#左对齐 ttk.Button(mainframe, text="选择图片", command=choosepic, style='ExtraLarge.TButton').grid(column=2, row=4, sticky=W) ttk.Button(mainframe, text="图像识别", command=btn_ai, style='ExtraLarge.TButton').grid(column=4, row=4, sticky=W) ttk.Button(mainframe, text="模型准确率", command=btn_acc, style='ExtraLarge.TButton').grid(column=6, row=4, sticky=W) ttk.Button(mainframe, text="关闭软件", command=btn_exit, style='ExtraLarge.TButton').grid(column=7, row=4, sticky=W) # 显示初始化 image_label = ttk.Label(root,compound=CENTER) image_label.grid(column=0,row=5, sticky=W) bg = "./bg2.PNG"#封面 pil_image = Image.open(bg)#打开图片 pil_image = pil_image.resize((1100, 700)) img = ImageTk.PhotoImage(pil_image) image_label.configure(image=img) root.mainloop()#进入Tkinter主事件循环，显示应用程序的界面并等待用户操作