Python OpenCV人脸识别、训练模型


1.引入库

代码如下(示例):

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

2.读入数据

代码如下(示例):

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

def train_img(path):
    #调用训练集
    face = cv.CascadeClassifier('haarcascade_frontalface_alt2.xml')
    #图片集
    picture = []
    #标签集
    ids = []
    #图片集合
    Imagepaths = [os.path.join(path, f) for f in os.listdir(path)]
    #循环每一张图片
    for Imagepath in Imagepaths:
        #提取这张图片的id号
        id = int(os.path.split(Imagepath)[1].split('.')[0])
        #将图片转灰
        PIL_img = Image.open(Imagepath).convert('L')
        #将图片转化为uint8
        img_numpy = np.array(PIL_img, 'uint8')
        #提取人脸
        faces = face.detectMultiScale(img_numpy)

        for x, y, w, h in faces:
            picture.append(img_numpy[y:y+h, x:x+w])
            ids.append(id)
    return picture, ids

path = 'D://ll//'
faces, ids = train_img(path)
recognizer = cv.face.LBPHFaceRecognizer_create()
recognizer.train(faces, np.array(ids))
recognizer.write('d:\\train.xml')

3.识别图片

import cv2 as cv
recognizer = cv.face.LBPHFaceRecognizer_create()
recognizer.read('d:\\train.xml')
def compare_img(img):
    face = cv.CascadeClassifier('haarcascade_frontalface_alt2.xml')
    gray = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
    faces = face.detectMultiScale(gray)
    for x, y, w, h in faces:
        cv.rectangle(img,(x, y), (x+w, y+h), (0,255,0),2)
        id, picture = recognizer.predict(gray[y:y+h, x:x+w])
        print('图片-->',id,'相似度评分:',picture)

    cv.imshow('out', img)
img = cv.imread('D:\\33.jpg')
compare_img(img)
cv.waitKey(0)
cv.destroyAllWindows()

总结

大功告成

物联沃分享整理
物联沃-IOTWORD物联网 » Python OpenCV人脸识别、训练模型

发表评论