|
|
|
@ -2,21 +2,19 @@ import threading
|
|
|
|
|
from tkinter import *
|
|
|
|
|
from tkinter import ttk
|
|
|
|
|
from PIL import Image, ImageTk
|
|
|
|
|
import numpy as np # 导入 numpy
|
|
|
|
|
import subprocess # 导入 subprocess 模块
|
|
|
|
|
from gesture_recognition import start_recognition, stop_recognition, release_camera, keep_running
|
|
|
|
|
import numpy as np
|
|
|
|
|
import subprocess
|
|
|
|
|
from gesture_recognition import start_recognition, stop_recognition, release_camera, keep_running, reset_hand_detection
|
|
|
|
|
|
|
|
|
|
WINDOW_WIDTH = 800
|
|
|
|
|
WINDOW_HEIGHT = 705
|
|
|
|
|
|
|
|
|
|
main_launched = False # 标志位,记录 main.py 是否已经启动
|
|
|
|
|
|
|
|
|
|
main_launched = False
|
|
|
|
|
status_label = None
|
|
|
|
|
|
|
|
|
|
def main():
|
|
|
|
|
"""显示欢迎屏幕"""
|
|
|
|
|
show_welcome_screen()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def set_window_position(window, width, height):
|
|
|
|
|
"""设置窗口位置居中"""
|
|
|
|
|
screen_width = window.winfo_screenwidth()
|
|
|
|
@ -25,7 +23,6 @@ def set_window_position(window, width, height):
|
|
|
|
|
y = (screen_height - height) // 2
|
|
|
|
|
window.geometry(f'{width}x{height}+{x}+{y}')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def show_welcome_screen():
|
|
|
|
|
"""显示欢迎界面"""
|
|
|
|
|
welcome = Tk()
|
|
|
|
@ -37,19 +34,17 @@ def show_welcome_screen():
|
|
|
|
|
bg_image = ImageTk.PhotoImage(bg_image)
|
|
|
|
|
|
|
|
|
|
background_label = Label(welcome, image=bg_image)
|
|
|
|
|
background_label.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 = 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, root, paused, popup_open, status_label
|
|
|
|
@ -86,7 +81,6 @@ def show_main_screen():
|
|
|
|
|
|
|
|
|
|
window.mainloop()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def start_thread(root, canvas, status_label):
|
|
|
|
|
"""启动识别线程"""
|
|
|
|
|
global keep_running, paused, popup_open
|
|
|
|
@ -94,16 +88,13 @@ def start_thread(root, canvas, status_label):
|
|
|
|
|
status_label.config(text="正在识别...")
|
|
|
|
|
paused = False
|
|
|
|
|
popup_open = False
|
|
|
|
|
threading.Thread(target=lambda: start_recognition(
|
|
|
|
|
callback=lambda fingers, img: root.after(0, update_canvas, canvas, fingers, img))).start()
|
|
|
|
|
|
|
|
|
|
threading.Thread(target=lambda: start_recognition(callback=lambda fingers, img: root.after(0, update_canvas, canvas, fingers, img))).start()
|
|
|
|
|
|
|
|
|
|
def stop_recognition_with_label(status_label):
|
|
|
|
|
"""停止识别并更新状态标签"""
|
|
|
|
|
stop_recognition()
|
|
|
|
|
status_label.config(text="已停止")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def exit_program():
|
|
|
|
|
"""退出程序"""
|
|
|
|
|
global window
|
|
|
|
@ -111,10 +102,9 @@ def exit_program():
|
|
|
|
|
release_camera()
|
|
|
|
|
window.destroy()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def update_canvas(canvas, fingers, img):
|
|
|
|
|
"""更新画布显示图像或处理手指数"""
|
|
|
|
|
if isinstance(img, np.ndarray): # 确保传入的img是图像数据
|
|
|
|
|
if isinstance(img, np.ndarray):
|
|
|
|
|
img = Image.fromarray(img)
|
|
|
|
|
imgtk = ImageTk.PhotoImage(image=img)
|
|
|
|
|
canvas.create_image(0, 0, anchor=NW, image=imgtk)
|
|
|
|
@ -122,7 +112,6 @@ def update_canvas(canvas, fingers, img):
|
|
|
|
|
canvas.update()
|
|
|
|
|
handle_finger_detection(fingers)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def handle_finger_detection(finger_count):
|
|
|
|
|
"""处理检测到的手指数"""
|
|
|
|
|
global paused, popup_open, main_launched
|
|
|
|
@ -135,31 +124,30 @@ def handle_finger_detection(finger_count):
|
|
|
|
|
popup_open = True
|
|
|
|
|
show_stop_recognition_window()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def launch_main_script():
|
|
|
|
|
"""启动 main.py 脚本"""
|
|
|
|
|
subprocess.Popen(['python', 'main.py'])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def show_stop_recognition_window():
|
|
|
|
|
"""显示停止识别确认窗口"""
|
|
|
|
|
|
|
|
|
|
def on_continue():
|
|
|
|
|
"""继续识别"""
|
|
|
|
|
global paused, popup_open
|
|
|
|
|
global paused, popup_open, status_label
|
|
|
|
|
paused = False
|
|
|
|
|
popup_open = False # 关闭弹窗后将标志设置为False
|
|
|
|
|
popup_open = False
|
|
|
|
|
stop_window.destroy()
|
|
|
|
|
reset_hand_detection()
|
|
|
|
|
start_thread(root, canvas, status_label)
|
|
|
|
|
|
|
|
|
|
def on_stop():
|
|
|
|
|
"""停止识别"""
|
|
|
|
|
global popup_open
|
|
|
|
|
stop_recognition()
|
|
|
|
|
popup_open = False # 关闭弹窗后将标志设置为False
|
|
|
|
|
popup_open = False
|
|
|
|
|
stop_window.destroy()
|
|
|
|
|
|
|
|
|
|
stop_window = Tk()
|
|
|
|
|
if root and root.winfo_exists():
|
|
|
|
|
stop_window = Toplevel(root)
|
|
|
|
|
stop_window.title("停止识别")
|
|
|
|
|
|
|
|
|
|
label = Label(stop_window, text="您竖起了五根手指,是否停止识别?", font=('Helvetica', 24, 'bold'))
|
|
|
|
@ -174,6 +162,5 @@ def show_stop_recognition_window():
|
|
|
|
|
stop_window.protocol("WM_DELETE_WINDOW", on_continue)
|
|
|
|
|
stop_window.mainloop()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
main()
|
|
|
|
|