时间戳

首页 > 解决方案 > 训练人脸时哪些参数会影响准确率，如何改进？

python - 训练人脸时哪些参数会影响准确率，如何改进？

问题描述

所以这是我的代码：

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

BASE_DIR = os.path.dirname(os.path.abspath(__file__))
image_dir = os.path.join(BASE_DIR, "images")

face_cascade = 
cv2.CascadeClassifier('cascades/data/haarcascade_frontalface_alt2.xml')
recognizer = cv2.face.LBPHFaceRecognizer_create()

current_id = 0
label_ids = {}
y_labels = []
x_train = []

for root, dirs, files in os.walk(image_dir):
    for file in files:
        if file.endswith("png") or file.endswith("jpg"):
            path = os.path.join(root, file)
            label = os.path.basename(root).replace(" ", "-").lower()
            #print(label, path)
            if not label in label_ids:
                label_ids[label] = current_id
                current_id += 1
            id_ = label_ids[label]  
            #print(label_ids)      
            #y_labels.append(label)# some number
            #x_train.append(path)# verify this image
            pil_image = Image.open(path).convert("L")
                size = (550, 550)
                final_image = pil_image.resize(size, Image.ANTIALIAS)
                image_array = np.array(final_image, "uint8")
                #print(image_array)
                faces = face_cascade.detectMultiScale(image_array, scaleFactor=1.5, minNeighbors=3)

            for (x, y, w, h) in faces:
                roi = image_array[y:y+h, x:x+w]
                x_train.append(roi)
                y_labels.append(id_)

当我开始训练存储在文件夹中的图像时，结果是并非所有图像都经过训练，识别人脸的准确率也不是那么好。

这里有一些参数我不太明白，但似乎对训练结果有很大影响。

有什么方法可以提高准确率吗？

这是我第一次提问，所以如果有类似的问题存在或没有输入正确的格式，请通知我。感谢您花时间阅读本文。

标签： pythonopencvcv2

解决方案

推荐阅读