SVM模型实现人脸识别

SVM训练过程

本文通过sklearn中的fetch_lfw_people数据集以及fetch_olivetti_faces数据集基于SVM模型分别训练人脸识别分类模型，进行简单的图像分类任务。

【fetch_lfw_people】数据集加载

from sklearn.decomposition import PCA
import numpy as np
from IPython.display import display
from sklearn.svm import SVC
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split
from sklearn.decomposition import PCA
from sklearn import datasets
from sklearn.model_selection import GridSearchCV


data = datasets.fetch_lfw_people(resize=1, min_faces_per_person=70)#70张人脸以上的样本

查看数据属性

display(dir(data))

数据基础属性

['DESCR', 'data', 'images', 'target', 'target_names']