一个基于预训练的DenseNet121模型的人脸年龄分类系统.zip

import time import tkinter as tk # gui用 import tkinter.messagebox # 弹出来的对话框 import torch import torchvision.models as models import torchvision.transforms as transforms import tkinter.filedialog # 文件相关窗口 from torch import nn from threading import Thread # 进程 import numpy as np from PIL import Image, ImageTk from mss import mss class MainWindow(tk.Tk): def __init__(self): super().__init__() self.default_path = [] # 默认模型位置,型配置位置和文件保存位置 self.title("年龄识别系统") # 给主窗口起一个名字 self.geometry("900x700+200+100") # 大小 self.config(menu=self.Generate_Menu()) # 生成菜单栏，窗口与菜单关联 self.now_num = -1 # 当前选择的功能 # 图片======================================================== self.image_ui = {} self.image_ui["图片框架"] = tk.LabelFrame(self, text="图片") # 图片显示 self.image_ui["滚动条x"] = tk.Scrollbar(self.image_ui["图片框架"], orient=tk.HORIZONTAL) # 滚动条x self.image_ui["滚动条y"] = tk.Scrollbar(self.image_ui["图片框架"], orient=tk.VERTICAL) # 滚动条y self.image_ui["画布"] = tk.Canvas(self.image_ui["图片框架"], xscrollcommand=self.image_ui["滚动条x"].set, yscrollcommand=self.image_ui["滚动条y"].set) self.image_ui["图片路径"] = "" self.image_ui["当前图片"] = Image.new('RGB', (0, 0)) self.image_ui["显示的图片"] = ImageTk.PhotoImage(self.image_ui["当前图片"]) self.image_ui["图片索引"] = self.image_ui["画布"].create_image(0, 0, anchor="nw", image=self.image_ui["显示的图片"]) # 先不放图片 self.image_ui["画布"].configure(scrollregion=(0, 0, 0, 0)) # 更新Canvas的滚动区域 self.image_ui["画布"].update() # 更新canvas以显示新图片 self.image_ui["滚动条x"].pack(side=tk.BOTTOM, fill=tk.X) # 靠下，拉满x self.image_ui["滚动条x"].config(command=self.image_ui["画布"].xview) self.image_ui["滚动条y"].pack(side=tk.RIGHT, fill=tk.Y) # 靠右，拉满y self.image_ui["滚动条y"].config(command=self.image_ui["画布"].yview) self.image_ui["画布"].pack(fill=tk.BOTH, expand=True) # 中间，且不扩充父框架大小 self.image_ui["图片框架"].grid(row=0, column=0, sticky="wesn") self.image_ui["画布"].bind('<Button-1>', self.Image_Click) # 绑定鼠标左键点击事件 self.image_ui["画布"].bind('<B1-Motion>', self.Image_Loosen) # 绑定鼠标左键点击移动事件 self.image_ui["画布"].bind('<ButtonRelease-1>', self.Image_Movex) # 绑定鼠标左键点击释放事件 self.image_ui["操作框架"] = tk.LabelFrame(self, text="操作") # 操作 self.image_ui["选择图片按钮"] = tk.Button(self.image_ui["操作框架"], text="选择图片", command=self.Select_Image) self.image_ui["图片大小标签"] = tk.Label(self.image_ui["操作框架"], text="图片大小") self.image_ui["图片大小滑条"] = tk.Scale(self.image_ui["操作框架"], from_=0.01, to=1, resolution=0.05, orient='horizontal', ) self.image_ui["图片大小滑条"].bind('<ButtonRelease-1>', self.Scale_Image) # 松开才执行 def Image_Change_Scale(c_num): self.image_ui["图片大小滑条"].set(self.image_ui["图片大小滑条"].get() + c_num) self.Scale_Image() self.image_ui["图片大小+"] = tk.Button(self.image_ui["操作框架"], text="+", command=lambda c_num=0.05: Image_Change_Scale(c_num)) self.image_ui["图片大小-"] = tk.Button(self.image_ui["操作框架"], text="-", command=lambda c_num=-0.05: Image_Change_Scale(c_num)) self.image_ui["显示的框选范围"] = [0, 0, 0] self.image_ui["真实的框选范围"] = [0, 0, 0] self.image_ui["框选大小标签"] = tk.Label(self.image_ui["操作框架"], text="框选大小") self.image_ui["框选大小滑条"] = tk.Scale(self.image_ui["操作框架"], from_=0, to=0, resolution=1, orient='horizontal', ) # 最大图片 def Image_Change_Selection_Box(event): if self.image_ui["框选大小滑条"].get() == self.image_ui["真实的框选范围"][2]: return else: self.image_ui["真实的框选范围"][2] = self.image_ui["框选大小滑条"].get() self.Show_Selection_Box() self.Update_Display() def Image_Change_Box_small(c_num=None, direction=None): if not c_num == None: self.image_ui["框选大小滑条"].set(self.image_ui["框选大小滑条"].get() + c_num) Image_Change_Selection_Box(None) elif not direction == None: # 8246,上下左右 num = self.image_ui["微调大小滑条"].get() if direction == 8: self.image_ui["真实的框选范围"][1] -= num if direction == 2: self.image_ui["真实的框选范围"][1] += num if direction == 4: self.image_ui["真实的框选范围"][0] -= num if direction == 6: self.image_ui["真实的框选范围"][0] += num self.Show_Selection_Box() self.Update_Display() self.image_ui["微调大小滑条"] = tk.Scale(self.image_ui["操作框架"], from_=1, to=100, resolution=1, orient='horizontal', ) self.image_ui["框选大小滑条"].bind('<ButtonRelease-1>', Image_Change_Selection_Box) # 松开才执行 self.image_ui["框选大小+"] = tk.Button(self.image_ui["操作框架"], text="+", command=lambda c_num=1: Image_Change_Box_small(c_num=c_num)) self.image_ui["框选大小-"] = tk.Button(self.image_ui["操作框架"], text="-", command=lambda c_num=-1: Image_Change_Box_small(c_num=c_num)) self.image_ui["微调上"] = tk.Button(self.image_ui["操作框架"], text="^", command=lambda: Image_Change_Box_small(direction=8)) self.image_ui["微调下"] = tk.Button(self.image_ui["操作框架"], text="v", command=lambda: Image_Change_Box_small(direction=2)) self.image_ui["微调左"] = tk.Button(self.image_ui["操作框架"], text="<", command=lambda: Image_Change_Box_small(direction=4)) self.image_ui["微调右"] = tk.Button(self.image_ui["操作框架"], text=">", command=lambda: Image_Change_Box_small(direction=6)) self.image_ui["示例图片框架"] = tk.LabelFrame(self.image_ui["操作框架"], text="输出示例") self.image_ui["示例图片画布"] = tk.Canvas(self.image_ui["示例图片框架"], width=224, height=224) self.image_ui["示例图片"] = Image.new('RGB', (224, 224)) self.image_ui["显示的示例图片"] = ImageTk.PhotoImage(self.image_ui["当前图片"]) self.image_ui["示例图片索引"] = self.image_ui["示例图片画布"].create_image(0, 0, anchor="nw", image=self.image_ui["显示的示例图片"]) self.image_ui["输出结果关联变量"] = tk.StringVar()