使用Tkinter和OpenCV训练人脸识别模型的问题

ia2d9nvy  于 2023-06-06  发布在  其他
关注(0)|答案(1)|浏览(176)

**描述:**我目前正在做一个人脸识别项目,使用Tkinter作为GUI,OpenCV在Visual Studio上进行图像处理。我的目标是用图像数据集训练模型,并显示一条指示训练完成的弹出消息。然而,我遇到了一个问题,当我点击“训练数据”按钮时,什么也没有发生。窗口未打开,也未显示错误消息。
编码

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")
        path=[os.path.join(data_dir,file) for  file in os.listdir(data_dir)]
        
        faces=[]
        ids=[]
        
        for image in  path:
            img=Image.open(image).convert('L')    #Gray Scale Image
            imageNp=np.array(img,'uint8')
            id=int(os.path.split(image)[1].split('.')[1])
            
            faces.append(imageNp)
            ids.append(id)  
            cv2.imshow("Training Data",imageNp)
            cv2.waitKey(1)==13
        
        ids=np.array(ids)
        
        #============Train the classifier and Save ===========
      
        clf=cv2.face.LBPHFaceRecognizer_create()
        clf.train(faces,ids)
        clf.write("Classifier.xml")
        messagebox.showinfo("Result","Training Datasets Completed!!")
        cv2.destroyAllWindows()

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

**预期行为:**当我点击“训练数据”按钮时,我希望打开一个窗口,显示训练进度,并显示图像。一旦训练完成,我希望弹出消息说“训练数据集完成!!”出现。分类器应保存为指定目录中的“Classifier.xml”。
**实际行为:**当我点击“TRAIN DATA”按钮时,没有任何React。窗口未打开,也未显示任何错误消息。未出现预期的指示培训完成的弹出消息。指定目录中未创建“Classifier.xml”文件
附加信息所有必需的依赖项都已安装。

pw9qyyiw

pw9qyyiw1#

尝试在不同的进程或线程中执行您的训练方法,并在线程完成时显示您的消息。我相信你的问题是你在GUI中使用了一个for循环

相关问题