parent
7b56ca6c34
commit
ee3c282d81
@ -0,0 +1,102 @@
|
||||
import threading
|
||||
from tkinter import *
|
||||
from tkinter import ttk
|
||||
from PIL import Image, ImageTk
|
||||
from gesture_recognition import start_recognition, stop_recognition, release_camera, keep_running
|
||||
|
||||
# 设置窗口宽高
|
||||
WINDOW_WIDTH = 800
|
||||
WINDOW_HEIGHT = 705
|
||||
|
||||
def main():
|
||||
show_welcome_screen()
|
||||
|
||||
def set_window_position(window, width, height):
|
||||
screen_width = window.winfo_screenwidth()
|
||||
screen_height = window.winfo_screenheight()
|
||||
x = (screen_width - width) // 2
|
||||
y = (screen_height - height) // 2
|
||||
window.geometry(f'{width}x{height}+{x}+{y}')
|
||||
|
||||
def show_welcome_screen():
|
||||
welcome = Tk()
|
||||
welcome.title("欢迎使用")
|
||||
set_window_position(welcome, WINDOW_WIDTH, WINDOW_HEIGHT)
|
||||
|
||||
# 加载背景图片
|
||||
bg_image = Image.open("1.png")
|
||||
bg_image = bg_image.resize((WINDOW_WIDTH, WINDOW_HEIGHT), Image.Resampling.LANCZOS)
|
||||
bg_image = ImageTk.PhotoImage(bg_image)
|
||||
|
||||
background_label = Label(welcome, image=bg_image)
|
||||
background_label.image = bg_image # 保持引用,防止垃圾回收
|
||||
background_label.place(relwidth=1, relheight=1)
|
||||
|
||||
label = Label(welcome, text="欢迎使用手势识别系统", font=('Helvetica', 24, 'bold'), bg="lightblue", fg="white")
|
||||
label.place(relx=0.5, rely=0.4, anchor=CENTER)
|
||||
|
||||
btn_continue = Button(welcome, text="继续", command=lambda: [welcome.destroy(), show_main_screen()], font=('Helvetica', 14), bg="yellow", fg="black", relief=SOLID)
|
||||
btn_continue.place(relx=0.5, rely=0.6, anchor=CENTER)
|
||||
|
||||
welcome.mainloop()
|
||||
|
||||
def show_main_screen():
|
||||
global window
|
||||
window = Tk()
|
||||
window.title("手势识别")
|
||||
set_window_position(window, WINDOW_WIDTH, WINDOW_HEIGHT)
|
||||
style = ttk.Style(window)
|
||||
style.theme_use('clam')
|
||||
|
||||
frame_controls = Frame(window, bg='lightgray', pady=10)
|
||||
frame_controls.pack(fill=X)
|
||||
|
||||
frame_video = Frame(window, bg='white')
|
||||
frame_video.pack(fill=BOTH, expand=True)
|
||||
|
||||
canvas = Canvas(frame_video, bg='black')
|
||||
canvas.pack(fill=BOTH, expand=True, padx=20, pady=20)
|
||||
|
||||
status_label = Label(frame_controls, text="等待开始...", font=('Helvetica', 14), bg='lightgray')
|
||||
status_label.pack(side=LEFT, padx=(10, 20))
|
||||
|
||||
btn_start = ttk.Button(frame_controls, text="开始", command=lambda: start_thread(canvas, status_label))
|
||||
btn_start.pack(side=LEFT, padx=10)
|
||||
|
||||
btn_stop = ttk.Button(frame_controls, text="停止", command=lambda: stop_recognition_with_label(status_label))
|
||||
btn_stop.pack(side=LEFT, padx=10)
|
||||
|
||||
btn_exit = ttk.Button(frame_controls, text="退出", command=lambda: exit_program())
|
||||
btn_exit.pack(side=LEFT, padx=10)
|
||||
|
||||
# 创建一个样式对象并设置按钮的样式
|
||||
style = ttk.Style(window)
|
||||
style.configure('TButton', font=('Helvetica', 14))
|
||||
|
||||
window.mainloop()
|
||||
|
||||
def start_thread(canvas, status_label):
|
||||
global keep_running
|
||||
if not keep_running:
|
||||
status_label.config(text="正在识别...")
|
||||
threading.Thread(target=lambda: start_recognition(callback=lambda img: update_canvas(canvas, img))).start()
|
||||
|
||||
def stop_recognition_with_label(status_label):
|
||||
stop_recognition()
|
||||
status_label.config(text="已停止")
|
||||
|
||||
def exit_program():
|
||||
global window
|
||||
stop_recognition()
|
||||
release_camera()
|
||||
window.destroy()
|
||||
|
||||
def update_canvas(canvas, img):
|
||||
img = Image.fromarray(img)
|
||||
imgtk = ImageTk.PhotoImage(image=img)
|
||||
canvas.create_image(320, 240, image=imgtk)
|
||||
canvas.image = imgtk
|
||||
canvas.update()
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
Loading…
Reference in new issue