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.
100 lines
3.4 KiB
100 lines
3.4 KiB
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, canvas
|
|
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()
|