用Python写一个简单的脸部识别项目(3)|训练数据并测试

代码实现

makeClassifier.py

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


recognizer =cv2.face.LBPHFaceRecognizer_create()
path='dataSet'

def getImagesWithID(path):
    imagePaths=[os.path.join(path,f) for f in os.listdir(path)]
    faces=[]
    IDs=[]
    usernames=[]
    for imagePath in imagePaths:
        faceImg=Image.open(imagePath).convert('L')
        faceNp=np.array(faceImg,'uint8')

        ID=int(os.path.split(imagePath)[-1].split('.')[1])
        username=os.path.split(imagePath)[-1].split('.')[2]

        faces.append(faceNp)
        IDs.append(ID)
        usernames.append(username)

        cv2.imshow("training",faceNp)
        cv2.waitKey(10)

    return np.array(IDs),faces,usernames

Ids,faces,usernames = getImagesWithID(path)

recognizer.train(faces, Ids)
recognizer.write('trainingData.yml')
cv2.destroyAllWindows()