OpenCV人脸识别EigenFace算法、案例解析-CSDN博客

本文链接：https://blog.csdn.net/m0_65065095/article/details/147992165

文章目录

前言
一、EigenFace 核心原理
二、Python 实战：手把手搭建 EigenFace 识别系统
三、优化技巧
四、总结

前言

在人脸识别领域，EigenFace 算法凭借主成分分析（PCA）的数学魅力，用简洁的逻辑实现高效的人脸特征提取。本文不仅深入剖析其原理，更通过 Python 代码带大家从 0 到 1 搭建 EigenFace 人脸识别系统。

一、EigenFace 核心原理

EigenFace 的核心是将高维人脸图像（比如 100×100 像素图像拉伸成 10000 维向量）映射到低维 “特征脸” 空间。
PCA（主成分分析）算法：
主成分分析（PCA）是一种矩阵的压缩算法，在减少矩阵维数的同时尽可能的保留原矩阵的信息，简单来说就是将 n×m的矩阵转换成n×k的矩阵，仅保留矩阵中所存在的主要特性，从而可以大大节省空间和数据量。
在这里插入图片描述
核心步骤如下：

数据预处理：灰度化并统一人脸图像尺寸，转化为一维向量构建训练集。
计算均值脸：算出所有训练图像的平均值，得到代表整体特征的 “平均脸”。
构建协方差矩阵：计算每张图像与平均脸的差异，构建协方差矩阵，挖掘数据分布规律。
特征分解：提取协方差矩阵的最大特征值对应的特征向量，这些就是关键的 “特征脸”。
人脸识别：将测试图像投影到特征脸空间，通过计算距离找到最匹配的训练样本。

二、Python 实战：手把手搭建 EigenFace 识别系统

我们使用 Python、OpenCV 和 NumPy 库，以 ORL 人脸数据集（40 人，每人 10 张图像）为例，逐步实现人脸识别系统。

1. 环境准备

pip install opencv-python numpy matplotlib

安装 OpenCV 处理图像，NumPy 进行数值计算，Matplotlib 用于可视化结果。

2. 代码实现与步骤详解

import cv2
import numpy as np
import os
from matplotlib import pyplot as plt

3. 加载数据集函数

def load_faces(path):
    faces = []  # 存储人脸图像向量
    labels = []  # 存储人脸标签
    label = 0  # 初始化标签

    for root, dirs, files in os.walk(path):  # 遍历数据集文件夹
        for file in files:  # 遍历每个文件
            img_path = os.path.join(root, file)  # 获取图像完整路径
            img = cv2.imread(img_path, cv2.IMREAD_GRAYSCALE)  # 以灰度模式读取图像
            faces.append(img.flatten())  # 将图像拉伸为一维向量并添加到列表
            labels.append(label)  # 添加对应标签
        label += 1  # 切换到下一个人的标签
    return np.array(faces), np.array(labels)  # 返回转换为NumPy数组的数据

4. 训练EigenFace模型函数

def train_eigenface(faces, labels):
    model = cv2.face.EigenFaceRecognizer_create()  # 创建EigenFace模型对象
    model.train(faces, labels)  # 使用训练数据训练模型
    return model  # 返回训练好的模型

5. 预测函数

def predict_face(model, test_face):
    label, confidence = model.predict(test_face)  # 对测试图像进行预测
    return label, confidence  # 返回预测标签和置信度

6.主程序部分

data_path = 'ORL_faces'  # 数据集路径
faces, labels = load_faces(data_path)  # 加载数据集

eigenface_model = train_eigenface(faces, labels)  # 训练模型

test_index = 50  # 选择一个测试图像索引
test_face = faces[test_index].reshape(1, -1)  # 取出测试图像并调整维度
predicted_label, confidence = predict_face(eigenface_model, test_face)  # 进行预测

print(f"预测标签: {predicted_label}, 置信度: {confidence}")  # 打印预测结果

7. 可视化结果

plt.subplot(1, 2, 1)
plt.imshow(faces[test_index].reshape(112, 92), cmap='gray')  # 显示测试图像
plt.title("测试图像")

plt.subplot(1, 2, 2)
plt.imshow(eigenface_model.getEigenVectors().T[0].reshape(112, 92), cmap='gray')  # 显示一个特征脸
plt.title("特征脸示例")
plt.show()  # 展示图像

8. 代码分步解读

数据加载：
load_faces函数通过os.walk遍历数据集文件夹，用cv2.imread读取灰度图像，再将图像拉伸为一维向量，同时生成对应的标签，最后以 NumPy 数组形式返回数据。
模型训练：
train_eigenface函数调用cv2.face.EigenFaceRecognizer_create()创建模型对象，传入训练数据完成训练，返回训练好的模型。
图像预测：
predict_face函数接收训练好的模型和测试图像，调用模型的predict方法得到预测标签和置信度。
结果展示：
主程序部分先加载数据、训练模型，再选取一张测试图像进行预测，最后用matplotlib可视化测试图像和一个特征脸，直观呈现算法效果。