万张PubFig人脸数据实现基于python+OpenCV的人脸特征定位程序

Posted 2022-11-23 itxiaoang

        在最近刷今日头条以及其他媒体软件时，经常会发现一些AI换脸的视频，于是我想，可不可以自己实现一个可以进行人脸识别的软件程序。我的具体流程是先配合python网络爬虫先进行万张PubFig人脸公共图片的爬取，分析出图片具体特征，然后再配合机器学习的OpenCV视觉库进行软件的构建。有一篇Github的文章讲得很详细，大家可以参考：https://github.com/Hironsan/BossSensor。

 

         前几篇博客先向大家讲解如何爬取PubFig人脸数据，然后本片的话先用一些动漫人脸图片，向大家展示基本的opencv库的操作，以及用一些公共人脸数据进行简单的人脸识别技术学习。

1.OpenCV简介

　　OpenCV是一个开源的跨平台计算机视觉库，提供的有python接口，并实现了图像处理和计算机视觉方面的很多通用算法。

 

2.需要安装的包

　　本篇先安装Opencv和numpy，pandas等数据分析包即可，如果感觉麻烦的话，可以直接安装Anaconda科学包(数据分析，挖掘，机器学习库合集)，安装与不同编译器配置环境过程这里就不讲解了。

　　下载地址(官网太慢了，推荐下面的地址)：https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple/opencv-python/。python3.5以上的需要下载opencv_python-4.0之类的版本，这里用到的是opencv_python-4.0.0.21-cp37-cp37m-win_amd64.whl这个文件。

　　安装完后，我们先用下面的代码输出一个图片：
　　

import cv2


img = cv2.imread("1.jpg", 1)
cv2.imshow("1", img)
cv2.waitKey()

结果如下：

发现我们的环境已经成功安装了。

 

3.初步人脸识别

由于本篇文章是第一篇，因此这里先简单的展示一下opencv的一些机器视觉的基础方法，下面就进行介绍。

(1)导入人脸图片，这里用一张动漫图

代码和上面的类似，更改图片即可：

import cv2


img = cv2.imread("firstPer.jpg", 1)
cv2.imshow("1", img)
cv2.waitKey()
cv2.destroyAllWindows()

这里我们把这段代码封装成一个函数：

def viewImage(image, name_of_window):
    ‘‘‘
    image:图像对象
    name_of_window:图像窗口名称
    ‘‘‘
    cv2.namedWindow(name_of_window, cv2.WINDOW_NORMAL)
    cv2.imshow(name_of_window, image)
    cv2.waitKey()
    cv2.destroyAllWindows()

(2)一些基础图像处理的方法

cropped：cropped = image[y:y+h, x:x+w]，就是以(y, x)为起点，裁剪大小为(h, w)的图像，以左上角为起点，竖直向下的方向为y轴，横向为x轴。

resized：dim = (width, height)

　　　　 resized = cv2.resize(img, dim, interpolation = cv2.INTER_AREA)

　　　　调整图像的大小。

用下面的代码分别进行这两种图像的处理：

img = cv2.imread("firstPer.jpg", 1)
# 裁剪图片大小，裁剪100范围的图片
cropped = img[0:100, 0:100]
viewImage(cropped,"firstPer")

img = cv2.imread("firstPer.jpg", 1)
scale_percent = 30    #调整30%的大小
width = int(img.shape[1] * scale_percent / 100)
height = int(img.shape[0] * scale_percent / 100)
dim = (width, height)
resized = cv2.resize(img, dim, interpolation = cv2.INTER_AREA)
viewImage(resized,"firstPer")

还有一一些用的操作方法，就是旋转，调节亮度，变模糊/平滑，绘制边框，绘制线段等，在这里并不是进行图像处理，因此就先不先向大家讲解。哈哈。

(3)图像灰度处理

       在进行人脸识别时，好多地方都先进行图像变灰度的操作，这里也给大家介绍一下：

这里有一个阈值函数，gray_image，将所有图像变为比127更暗直至0或者增加亮度到255，将图像的彩色边框的内容略过。

灰度处理代码：

gray_image = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
viewImage(gray_image,"gray Image")

结果：

亮度彩色阈值处理代码：

one,threshold_image = cv2.threshold(img, 127, 255, 0)
viewImage(threshold_image,"firstPer")

结果：

主要是有后面的三个参数控制，大家有兴趣可以自行学习。

(4)人脸识别初讲

　　如果我们需要进行人脸识别，需要下载一些opencv配置文件，地址为：https://github.com/opencv/opencv/tree/master/data/haarcascades，这里我们用到的是haarcascade_frontalface_default.xml文件。用下面的代码进行检测：

face_cascade = cv2.CascadeClassifier(‘haarcascade_frontalface_default.xml‘)
img = cv2.imread("firstPer.jpg")
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
faces = face_cascade.detectMultiScale(
gray,                # 灰度图
scaleFactor= 1.1,    # 缩放比例
minNeighbors= 2,    # 识别出一个人脸之前在当前物体周围需要检测的物体数目
minSize=(50, 50)    #窗口的大小
)

firstPer = format(len(faces)) + "faces detected!"
print(firstPer)# Draw a rectangle around the faces
for (x, y, w, h) in faces:
    cv2.rectangle(img, (x, y), (x+w, y+h), (255, 255, 0), 2)
viewImage(img,firstPer)

detectMultiScale函数是一个检测物体的通用函数。当我们把它用于人脸检测时，它就会从图像中检测出人脸。

但是我们却发现，根本没有检测出我们可爱的小樱的脸，这是为什么呢？不急，我们先用其他图片检测：

再来一张最美的图片：

       这时我们发现，识别程序识别的时候会有两个要注意的地方，第一个是动漫图片线条简单，有时候并不能识别出；而且识别的能力和图片的清晰度也有关系。

       针对第一个问题，在接下来的教程中，我会使用哥伦比亚大学的公共PubFig人脸库作为人脸识别数据集，进行机器视觉的训练。向大家讲解一下如何用机器学习训练一个不仅仅能够识别人脸，而且还能够识别表情的软件，欢迎大家继续关注我的博客，如果有宝贵建议的话，请在下方评论。