无法使用LBPHFaceRecognitor和opencv训练图像

2024-05-18 17:52:14 发布

您现在位置:Python中文网/ 问答频道 /正文

目前,我正在为人脸识别系统培训图像。我正在使用Python、OpenCV来实现这一点。我从网络摄像头中收集了样本,但是样本图像的大小不同。示例图像的大小为376 x 376、412 x 412、836 x 836

当前工作目录的屏幕截图:

enter image description here

样本图像保存在名为“sampleImgFolder”的主文件夹中,并保存在每个样本的主文件夹特定文件夹下

enter image description here


训练图像的源代码


import os
import cv2
import numpy as np
from PIL import Image

recognizer = cv2.face.LBPHFaceRecognizer_create()
targetImagesDirectory="sampleImgFolder/"
dataset = cv2.CascadeClassifier('resources/haarcascade_frontalface_default.xml')

def getImageWithID(path):
    #empty list to store processed data 
    sampleFaces = []
    sampleFaceId = []

    os.chdir(targetImagesDirectory)
    for directory in os.listdir():
        os.chdir(directory)
        for files in os.listdir():
            imagePath = '{}/{}'.format(os.getcwd(), files)

            imagePil = Image.open(imagePath).convert('L')
            imageNumpy = np.array(imagePil, 'uint8') #conversion of normal image to numpy array
            #imageNumpy.astype(np.float32)

            #detect face 
            faces = dataset.detectMultiScale(imageNumpy)

            #extracting id from file name
            id = files.split('_')
            id = id[0].split('-')
            id = id[2]

            for (x, y, w, h) in faces:
                sampleFaces.append(imageNumpy[y:y + h, x:x + w])
                sampleFaceId.append(id)

        os.chdir('../')
    os.chdir('../')   

    return np.array(sampleFaceId), sampleFaces
print("reading images")
Ids,faces=getImageWithID(targetImagesDirectory)
print('reading completed')
recognizer.train(faces,Ids)
print("training")
#train the dataset. Create a file name trainningData.yml
recognizer.write('train/trainningData.yml')
cv2.destroyAllWindows()

运行上述代码时,我遇到以下错误:

enter image description here


Tags: 图像import文件夹idosnpcv2dataset

热门问题