使用Tkinter与OpenCV训练人脸识别模型时按钮无响应问题

核心问题排查 #

点击按钮无响应的本质是代码执行过程中出现未捕获的异常，而Tkinter回调函数默认会静默处理异常，导致无任何反馈。结合代码来看，主要触发点包括：

• 训练数据目录images data不存在或路径错误
• 图像文件名不符合预期格式（比如缺少第二个.分割符）
• OpenCV窗口显示逻辑阻塞Tkinter主线程
• 未对异常进行捕获与提示

修复步骤与代码调整 #

1. 增加异常捕获与错误提示 #

在训练方法中加入异常捕获，通过messagebox展示错误信息，方便定位问题。

2. 验证数据目录存在 #

遍历文件前先检查目标目录是否存在，不存在则直接提示用户。

3. 修正文件名解析逻辑 #

增加文件名格式校验，避免因格式不符导致索引越界错误。

4. 优化OpenCV窗口显示逻辑 #

如果不需要实时预览图像，可直接删除相关代码避免阻塞主线程；若保留预览，需确保按键逻辑正确。

修正后的完整代码 #

from tkinter import *
from tkinter import ttk
from PIL import Image, ImageTk
from tkinter import messagebox
import mysql.connector
import cv2
import os
import numpy as np

class Train:
    def __init__(self, root):
        self.root = root
        self.root.geometry("1530x790+0+0")
        self.root.title("Face Recognition System")  
        
        title_lbl= Label(self.root, text= "TRAIN DATA SET",font=("times new roman",35,"bold"), bg="white",fg="darkgreen")
        title_lbl.place(x=0,y=0,width=1530, height=45)
        
        img_top= Image.open(r"C:\Users\The-Javeira\Desktop\test\images\b.jpg")
        img_top= img_top.resize((1530,325), Image.ANTIALIAS)
        self.photoimg_top= ImageTk.PhotoImage(img_top)
        
        f_lbl2= Label(self.root,image=self.photoimg_top)
        f_lbl2.place(x=0,y=55,width=1530,height=325)  
        
        #button
        b1_1=Button(self.root ,text="TRAIN DATA", command=self.train_classifier, cursor="hand2",font=("times new roman",30,"bold"), bg="red",fg="white")
        b1_1.place(x=0,y=380,width=1530,height=60)
        
        img_bottom= Image.open(r"C:\Users\The-Javeira\Desktop\test\images\train2.JPEG")
        img_bottom= img_bottom.resize((1530,325), Image.ANTIALIAS)
        self.photoimg_bottom= ImageTk.PhotoImage(img_bottom)
        
        f_lbl2= Label(self.root,image=self.photoimg_bottom)
        f_lbl2.place(x=0,y=440 ,width=1530,height=325)  
        
    def train_classifier(self):
        data_dir = "images data"
        # 检查数据目录是否存在
        if not os.path.exists(data_dir):
            messagebox.showerror("Error", f"目录 {data_dir} 不存在，请检查路径！")
            return
        
        try:
            # 仅遍历图像格式文件
            path = [os.path.join(data_dir, file) for file in os.listdir(data_dir) if file.endswith(('.png', '.jpg', '.jpeg'))]
            
            faces = []
            ids = []
            
            for image_path in path:
                # 处理灰度图像
                img = Image.open(image_path).convert('L')
                imageNp = np.array(img, 'uint8')
                
                # 解析ID并校验格式
                filename = os.path.split(image_path)[1]
                parts = filename.split('.')
                if len(parts) < 2:
                    messagebox.showwarning("警告", f"文件名 {filename} 格式不符合预期，跳过该文件")
                    continue
                try:
                    id = int(parts[1])
                except ValueError:
                    messagebox.showwarning("警告", f"文件名 {filename} 中的ID不是数字，跳过该文件")
                    continue
                
                faces.append(imageNp)
                ids.append(id)
                
                # 实时预览训练图像，按ESC可提前退出预览
                cv2.imshow("Training Data", imageNp)
                if cv2.waitKey(1) & 0xFF == 27:
                    break
            
            # 检查是否有有效训练数据
            if not faces:
                messagebox.showwarning("提示", "没有有效训练数据，请检查数据集！")
                cv2.destroyAllWindows()
                return
            
            ids = np.array(ids)
            
            # 训练并保存分类器
            clf = cv2.face.LBPHFaceRecognizer_create()
            clf.train(faces, ids)
            clf.write("Classifier.xml")
            
            messagebox.showinfo("Result","Training Datasets Completed!!")
            cv2.destroyAllWindows()
            
        except Exception as e:
            messagebox.showerror("错误", f"训练过程中出现异常：{str(e)}")
            cv2.destroyAllWindows()


if __name__ == "__main__":
    root = Tk()
    obj = Train(root)
    root.mainloop()

额外注意事项 #

• 确保images data目录位于脚本运行目录下，或使用绝对路径（如data_dir = r"C:\Users\The-Javeira\Desktop\test\images data"）
• 图像文件名需符合xxx.数字.xxx格式（如user.1.jpg），否则会被跳过或触发警告
• 若无需实时预览，可删除cv2.imshow和cv2.waitKey相关代码，避免阻塞Tkinter主线程

内容的提问来源于stack exchange，提问作者javeria saeed