温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

如何用python创建人脸识别程序

发布时间:2021-12-27 14:10:12 来源:亿速云 阅读:240 作者:iii 栏目:大数据

这篇文章主要讲解了“如何用python创建人脸识别程序”,文中的讲解内容简单清晰,易于学习与理解,下面请大家跟着小编的思路慢慢深入,一起来研究和学习“如何用python创建人脸识别程序”吧!

Facenet

在这个项目中,我们将使用一个名为Facenet的系统来为我们做人脸识别程序。

 Facenet是什么?

Facenet是由Florian Schroff、Dmitry Kalenichenko和James Philbin构建的系统。他们也为此写了一篇论文。

它直接从人脸图像中学习图像到欧式空间上点的映射,其中距离直接对应于人脸相似度的度量。一旦创建了这些嵌入,就可以使用这些嵌入作为特征来完成人脸识别和验证等过程。

 Facenet是如何工作的?

Facenet使用卷积层直接从人脸的像素学习。该网络在一个大数据集上进行训练,以实现对光照、姿态和其他可变条件的不变性。该系统是在 Labelled Faces in the wild(http://vis-www.cs.umass.edu/lfw/) 数据集上训练的。这个数据集包含13000多张从网络上收集的不同人脸的图片,每个人脸都有一个名字(标签)。

Facenet从图像中创建128维嵌入,并将其插入特征空间,这样,无论成像条件如何,相同身份之间的特征距离要尽可能的小,而不同身份之间的特征距离要尽可能的大。下图描述了模型体系结构:

如何用python创建人脸识别程序

模型结构:模型包含一个批处理输入层、一个深入学习架构和一个L2层。这就产生了人脸嵌入。

 三重损失

该系统采用一种特殊的损失函数,称为三重损失。三重损失使同一身份的图像之间的L2距离最小,使不同特征的人脸图像之间的L2距离最大。

该系统采用三重损失,更适合于人脸验证。使用三重损失的动机是它鼓励将一个身份的所有图像投影到嵌入空间中的一个点上。

三重损失:学习前后 如何用python创建人脸识别程序

创作者们设计了一种高效的三重选择机制,可以一次巧妙地选择三幅图像。这些图像有以下三种类型:

  1. 锚:一个随机的人的图。

  2. 正图:同一个人的另一张图。

  3. 负图:另一个人的图。

测量了两个欧几里德距离:一个是锚和正图之间的距离,我们称之为A;另一个是锚和负图之间的距离,我们称之为B。训练过程旨在减少A并使B最大化,这样相似的图像彼此靠近,不同的图像在嵌入空间中会隔得很远。

 开始构建!

最精彩的部分开始了!我们可以使用Facenet为我们自己选择的人脸创建嵌入,然后训练支持向量机(Support Vector Machine)使用这些嵌入并进行分类。让我们开始建立一个自定义的人脸识别程序吧!

你可以看看这个项目的Github存储库,因为它包含了一个自定义数据集和用于在视频中检测人脸的脚本。

 先决条件

在开始之前,请确保你的系统上安装了以下库:

  1. tensorflow==1.7

  2. scipy

  3. scikit-learn

  4. opencv-python

  5. h6py

  6. matplotlib

  7. Pillow

  8. requests

  9. psutil

 下载数据

在这个项目中,我们将创建一个人脸识别程序,将能够识别90年代情景喜剧《老友记》的核心人物。如果你想识别一组不同的人脸,那么就用你的图像来代替。只需确保遵循类似的目录结构——为要识别的每个身份创建一个文件夹,并将这些文件夹存储在名为“raw”的文件夹中。

数据集目录:注意每个角色如何拥有自己对应的文件夹

如何用python创建人脸识别程序

在每个文件夹里放上人物的照片。请注意,每张图片只有一张清晰可见的脸。尽管只需要很少的图像,我还是为每个角色添加了20个图像。每个文件夹都有相同数量的图片。你可以从这里下载我创建的《老友记》的数据集。顺便说一下,这是“Chandler”文件夹的样子:

如何用python创建人脸识别程序

 下载Facenet

现在已经结束收集数据了。请继续下载Facenet repo。下载并提取它,并将“Dataset”文件夹放入其中。

 对齐

该模型的一个问题是它可能会漏掉一些人脸标志,为了解决这个问题,我们必须将数据集中的所有图像对齐,使眼睛和嘴唇在所有图片中显示在同一位置。我们将使用M.T.C.N.N.(多任务C.N.N.)来执行相同的操作,并将所有对齐的图像存储在名为processed的文件夹中。

打开终端/命令提示符并导航到Facenet目录。然后运行align_dataset_mtcn.py以及以下参数。

python src/align_dataset_mtcnn.py \
./Dataset/Friends/raw \
./Dataset/Friends/processed \
--image_size 160 \
--margin 32 \
--random_order \ 
--gpu_memory_fraction 0.25

运行此命令将对齐所有图像并将其存储在各自的文件夹中,然后将所有内容存储在“processed”文件夹中。下图将向你介绍对齐的工作原理:

如何用python创建人脸识别程序

所有图像都被裁剪并与标准的160x160像素图像对齐。

 获得预先训练的模型

现在,为了在你自己的图像上训练模型,你需要下载预先训练的模型。

在Facenet根目录中创建一个名为“Models”的文件夹。下载完成后,将zip文件的内容解压缩到名为“facenet”的目录中,并将此文件夹放在“Models”文件夹中。

如何用python创建人脸识别程序

这个模型是在LFW数据集上训练的,因此所有的人脸嵌入都存储在这些文件中。这使我们有机会冻结图像,并在我们自己的图像上训练它。这样做会将我们提供的所有人脸嵌入到维度空间中。

 用我们的数据训练模型

我们都准备好了!我们有一个预先训练好的模型,我们的自定义数据集已经对齐并准备好了。现在,是时候训练模型了!

python src/classifier.py TRAIN \
./Dataset/Friends/processed \
./Models/facenet/20180402-114759.pb \
./Models/Friends/Friends.pkl \
--batch_size 1000

执行上述命令将加载预先训练的模型并启动训练过程。训练结束后,新图像的嵌入将导出到/Models/Friends/中。

由于我们使用的是预先训练的模型和相对较少的图像数量,因此训练过程很快就结束了。

 在视频源上测试我们的模型

为了测试我们的模型,我使用的是来自《老友记》的视频。你可以用自己的视频来代替,甚至可以用摄像头。在本节中,我们将编写脚本,以便于在视频源中进行人脸识别。

导航到“src”文件夹并创建一个新的python脚本。我给它起名 faceRec.py。

接下来,我们导入所有必需的库。

此脚本只接受一个参数,即视频文件的路径。如果没有提到路径,那么我们将通过网络摄像头传输视频。因此,参数的默认值为0。

我们将初始化一些变量。请确保根据文件夹结构更改路径。

加载自定义分类器。

设置Tensorflow图,然后加载Facenet模型。使用GPU将加快检测和识别过程。

设置输入和输出张量。

pnet、rnet和onet是M.T.C.N.N.的组成部分,将用于检测和对齐人脸。

接下来,我们将创建一个集合和一个来跟踪检测到的每个角色的集合。

设置视频捕获对象。

因此,如果在运行程序时未将VIDEO_PATH作为参数传递,则它将假定默认值为0。如果发生这种情况,视频捕获对象将从网络摄像机流式传输视频。

然后逐帧捕获视频,并且由检测人脸模块在这些帧中检测人脸。找到的人脸数存储在faces\u found变量中。

如果找到人脸,那么我们将迭代每个人脸并将边界框的坐标保存在变量bb中。

然后提取、裁剪、缩放、重塑这些人脸并输入字典。

我们将使用该模型来预测人脸的身份。我们提取最佳类概率或置信度。这是衡量我们的模型是如何确定预测的身份属于给定的脸。

最后,我们将在人脸周围画一个边界框,并在边界框旁边写下预测的身份和置信度。如果置信度低于某个阈值,我们将把名字填为未知。

一定要放一个except语句。这将确保成功忽略抛出的任何错误。确保放置except语句。这样做有助于我们忽略错误。

except: pass

显示视频并在过程结束后关闭视频显示窗口。因为每一帧都要经过大量的处理,所以视频回放可能会很慢。

恭喜你,你的耐心得到了回报!我们已经完成了脚本,准备好了!快速启动并执行以下命令以启动人脸识别程序。请确保将要测试的视频的路径作为参数传递,或将其留空以从网络摄像机流式传输视频。

python src/faceRec.py --path ./Dataset/Friends/friends.mp

如何用python创建人脸识别程序

好吧,这个系统还不完善,还有一些缺点。

 缺点
  1. 系统总是试图将每个人脸都匹配到一个给定的身份中。如果屏幕上出现新人脸,系统将为其分配一个或另一个身份。这个问题可以通过仔细选择一个阈值来解决。

  2. 身份的混淆。在上面的gif中,你可以观察到Joey和Chandler之间的预测有时是如何波动的。而且,置信度得分也很低。使用更多图像训练模型将解决此问题。

  3. 无法在一定距离识别人脸(如果距离很远使得人脸看起来很小)。

感谢各位的阅读,以上就是“如何用python创建人脸识别程序”的内容了,经过本文的学习后,相信大家对如何用python创建人脸识别程序这一问题有了更深刻的体会,具体使用情况还需要大家实践验证。这里是亿速云,小编将为大家推送更多相关知识点的文章,欢迎关注!

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI