50行Python代码实现人脸检测功能


Posted in Python onJanuary 23, 2018

50行Python代码实现人脸检测功能 

现在的人脸识别技术已经得到了非常广泛的应用,支付领域、身份验证、美颜相机里都有它的应用。用iPhone的同学们应该对下面的功能比较熟悉

50行Python代码实现人脸检测功能 

iPhone的照片中有一个“人物”的功能,能够将照片里的人脸识别出来并分类,背后的原理也是人脸识别技术。

这篇文章主要介绍怎样用Python实现人脸检测。人脸检测是人脸识别的基础。人脸检测的目的是识别出照片里的人脸并定位面部特征点,人脸识别是在人脸检测的基础上进一步告诉你这个人是谁。

好了,介绍就到这里。接下来,开始准备我们的环境。

准备工作

本文的人脸检测基于dlib,dlib依赖Boost和cmake,所以首先需要安装这些包,以Ubuntu为例:

$ sudo apt-get install build-essential cmake 
$ sudo apt-get install libgtk-3-dev 
$ sudo apt-get install libboost-all-dev

我们的程序中还用到numpy,opencv,所以也需要安装这些库:

$ pip install numpy 
$ pip install scipy 
$ pip install opencv-python 
$ pip install dlib

人脸检测基于事先训练好的模型数据,从这里可以下到模型数据

http://dlib.net/files/shape_predictor_68_face_landmarks.dat.bz2

下载到本地路径后解压,记下解压后的文件路径,程序中会用到。

dlib的人脸特征点

上面下载的模型数据是用来估计人脸上68个特征点(x, y)的坐标位置,这68个坐标点的位置如下图所示

50行Python代码实现人脸检测功能 

我们的程序将包含两个步骤:

第一步,在照片中检测人脸的区域

第二部,在检测到的人脸区域中,进一步检测器官(眼睛、鼻子、嘴巴、下巴、眉毛)

人脸检测代码

我们先来定义几个工具函数:

def rect_to_bb(rect): 
  x = rect.left() 
  y = rect.top() 
  w = rect.right() - x 
  h = rect.bottom() - y   
  return (x, y, w, h)

这个函数里的rect是dlib脸部区域检测的输出。这里将rect转换成一个序列,序列的内容是矩形区域的边界信息。

def shape_to_np(shape, dtype="int"): 
  coords = np.zeros((68, 2), dtype=dtype)   
  for i in range(0, 68): 
      coords[i] = (shape.part(i).x, shape.part(i).y)   
  return coords

这个函数里的shape是dlib脸部特征检测的输出,一个shape里包含了前面说到的脸部特征的68个点。这个函数将shape转换成Numpy array,为方便后续处理。

def resize(image, width=1200): 
  r = width * 1.0 / image.shape[1] 
  dim = (width, int(image.shape[0] * r)) 
  resized = cv2.resize(image, dim, interpolation=cv2.INTER_AREA)   
  return resized

这个函数里的image就是我们要检测的图片。在人脸检测程序的最后,我们会显示检测的结果图片来验证,这里做resize是为了避免图片过大,超出屏幕范围。

接下来,开始我们的主程序部分

import sys import numpy as np 
import dlib import cv2 
if len(sys.argv) < 2:   
  print "Usage: %s <image file>" % sys.argv[0] 
  sys.exit(1) 
image_file = sys.argv[1] 
detector = dlib.get_frontal_face_detector() 
predictor = dlib.shape_predictor("shape_predictor_68_face_landmarks.dat")

我们从sys.argv[1]参数中读取要检测人脸的图片,接下来初始化人脸区域检测的detector和人脸特征检测的predictor。shape_predictor中的参数就是我们之前解压后的文件的路径。

image = cv2.imread(image_file) 
image = resize(image, width=1200) 
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) 
rects = detector(gray, 1)

在检测特征区域前,我们先要检测人脸区域。这段代码调用opencv加载图片,resize到合适的大小,转成灰度图,最后用detector检测脸部区域。因为一张照片可能包含多张脸,所以这里得到的是一个包含多张脸的信息的数组rects。

for (i, rect) in enumerate(rects): 
  shape = predictor(gray, rect) 
  shape = shape_to_np(shape) 
  (x, y, w, h) = rect_to_bb(rect) 
  cv2.rectangle(image, (x, y), (x + w, y + h), (0, 255, 0), 2) 
  cv2.putText(image, "Face #{}".format(i + 1), (x - 10, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)   
for (x, y) in shape: 
      cv2.circle(image, (x, y), 2, (0, 0, 255), -1) 
cv2.imshow("Output", image) 
cv2.waitKey(0)

对于每一张检测到的脸,我们进一步检测脸部的特征(鼻子、眼睛、眉毛等)。对于脸部区域,我们用绿色的框在照片上标出;对于脸部特征,我们用红色的点标出来。

最后我们把加了检测标识的照片显示出来,waitKey(0)表示按任意键可退出程序。

以上是我们程序的全部

测试

接下来是令人兴奋的时刻,检验我们结果的时刻到来了。

下面是原图

50行Python代码实现人脸检测功能 

下面是程序识别的结果

50行Python代码实现人脸检测功能 

可以看到脸部区域被绿色的长方形框起来了,脸上的特征(鼻子,眼睛等)被红色点点标识出来了。

总结

以上所述是小编给大家介绍的50行Python代码实现人脸检测功能,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对三水点靠木网站的支持!

Python 相关文章推荐
python迭代器实例简析
Sep 25 Python
python使用Apriori算法进行关联性解析
Dec 21 Python
Python内置模块logging用法实例分析
Feb 12 Python
使用Python搭建虚拟环境的配置方法
Feb 28 Python
pandas全表查询定位某个值所在行列的方法
Apr 12 Python
解决Python安装后pip不能用的问题
Jun 12 Python
详解如何将python3.6软件的py文件打包成exe程序
Oct 09 Python
Python使用while循环花式打印乘法表
Jan 28 Python
Python Web框架之Django框架Model基础详解
Aug 16 Python
python通过robert、sobel、Laplace算子实现图像边缘提取详解
Aug 21 Python
Python命令行参数解析工具 docopt 安装和应用过程详解
Sep 26 Python
python实现简单的井字棋
May 26 Python
Python基于OpenCV实现视频的人脸检测
Jan 23 #Python
Python求出0~100以内的所有素数
Jan 23 #Python
python之matplotlib学习绘制动态更新图实例代码
Jan 23 #Python
彻底搞懂Python字符编码
Jan 23 #Python
Python实现PS滤镜的万花筒效果示例
Jan 23 #Python
python处理csv数据动态显示曲线实例代码
Jan 23 #Python
Python+matplotlib实现华丽的文本框演示代码
Jan 22 #Python
You might like
非洲第一个咖啡超凡杯大赛承办国—卢旺达的咖啡怎么样
2021/03/03 咖啡文化
php判断输入不超过mysql的varchar字段的长度范围
2011/06/24 PHP
php输出xml格式字符串(用的这个)
2012/07/12 PHP
PHP代码保护--Zend Guard的使用详解
2013/06/03 PHP
使用php测试硬盘写入速度示例
2014/01/27 PHP
个人写的PHP验证码生成类分享
2014/08/21 PHP
php生成高清缩略图实例详解
2015/12/07 PHP
php版银联支付接口开发简明教程
2016/10/14 PHP
PHP解决中文乱码
2017/04/28 PHP
Jquery之Ajax运用 学习运用篇
2011/09/26 Javascript
JavaScript实现继承的4种方法总结
2014/10/16 Javascript
使表格的标题列可左右拉伸jquery插件封装
2014/11/24 Javascript
jQuery给动态添加的元素绑定事件的方法
2015/03/09 Javascript
Bootstrap实现响应式导航栏效果
2015/12/28 Javascript
JS深度拷贝Object Array实例分析
2016/03/31 Javascript
jquery插件格式实例分析
2016/06/16 Javascript
javascript 广告移动特效的实现代码
2016/06/25 Javascript
Angularjs中的ui-bootstrap的使用教程
2017/02/19 Javascript
ES6新特性之Symbol类型用法分析
2017/03/31 Javascript
微信小程序上传多图到服务器并获取返回的路径
2019/05/05 Javascript
element el-table表格的二次封装实现(附表格高度自适应)
2021/01/19 Javascript
对numpy.append()里的axis的用法详解
2018/06/28 Python
python 定义n个变量方法 (变量声明自动化)
2018/11/10 Python
在python带权重的列表中随机取值的方法
2019/01/23 Python
win8.1安装Python 2.7版环境图文详解
2019/07/01 Python
基于Python2、Python3中reload()的不同用法介绍
2019/08/12 Python
python实现回旋矩阵方式(旋转矩阵)
2019/12/04 Python
python如何快速拼接字符串
2020/10/28 Python
Python爬虫制作翻译程序的示例代码
2021/02/22 Python
会计专业自我鉴定
2014/02/10 职场文书
物流管理专业毕业生自荐信
2014/03/04 职场文书
同居协议书范本
2014/04/23 职场文书
八荣八耻演讲稿
2014/09/15 职场文书
辞职书格式样本
2015/02/26 职场文书
同事欢送会致辞
2015/07/31 职场文书
python scrapy简单模拟登录的代码分析
2021/07/21 Python