码迷,mamicode.com
首页 > 编程语言 > 详细

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

时间:2019-09-22 17:57:04      阅读:307      评论:0      收藏:0      [点我收藏+]

标签:窗口   width   rop   min   爬虫   裁剪   就是   ast   参数   

        在最近刷今日头条以及其他媒体软件时,经常会发现一些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人脸库作为人脸识别数据集,进行机器视觉的训练。向大家讲解一下如何用机器学习训练一个不仅仅能够识别人脸,而且还能够识别表情的软件,欢迎大家继续关注我的博客,如果有宝贵建议的话,请在下方评论。

 

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

标签:窗口   width   rop   min   爬虫   裁剪   就是   ast   参数   

原文地址:https://www.cnblogs.com/ITXiaoAng/p/11553402.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!