YOLO and DarkNet
Face recognition based on YOLO, You Only Look Once: Unified, Real-Time Object Detection.
# 1. Abstract
自 YOLO 算法提出以来,至今已经发展到了 v3,性能、集成性等都得到了极大的提升,用 YOLO 来实现人脸识别算法,其特点是模型训练参数较少,可移植并且实时性很高。目前为止,集成现有技术实现一个基于 YOLO 算法的人脸识别系统是一项很有挑战性的工作。近几年来,目标检测算法取得了很大的突破。比较流行的算法可以分为两类,一类是基于 Region Proposal 的 R-CNN 系算法(R-CNN,Fast R-CNN, Faster R-CNN),它们是 two-stage 的,需要先使用启发式方法(selective search)或者 CNN 网络(RPN)产生 Region Proposal,然后再在 Region Proposal 上做分类与回归。另一类是 Yolo,SSD 这类 one-stage 算法,其仅仅使用一个 CNN 网络直接预测不同目标的类别与位置。第一类方法准确度高,但是速度慢,第二类算法速度快,但是准确性较低。本文将介绍 Yolo 算法,其全称是 You Only Look Once: Unified, Real-Time Object Detection,You Only Look Once 说的是只需要一次 CNN 运算,Unified 指的是这是一个统一的框架,提供 end-to-end 的预测,而 Real-Time 体现是 Yolo 算法速度快。这里我们谈的是 Yolo-v1 版本算法,其性能差于后来的 SSD 算法的,但是 Yolo 后来也继续进行改进,产生了 Yolo9000 算法。本文主要讲述 Yolo-v1 算法的原理。
# 2. YOLO
# 2.1 Sliding window
The sliding window model is conceptually simple: independently classify all image patches as being object or non-object. Sliding window classification is the dominant paradigm in object detection and for one object category in particular -- faces -- it is one of the most noticeable successes of computer vision. For example, modern cameras and photo organization tools have prominent face detection capabilities.
如引文所示,常见的目标检测算法一般基于滑动窗口模型,其思路是,将检测问题转化为了图像分类问题。其基本原理就是采用不同大小和窗口在整张图片上以一定的步长进行滑动,然后对这些窗口对应的区域做图像分类,以此实现对整张图片的检测。
但该方法的缺点在于,要检测的目标大小规模是未知的,所以在实验中需要设置不同大小的窗口去滑动,并且还要选取合适的步长。这样的做法会产生很多的子区域经过分类器去做预测,需要很大的计算量,但是为了保证速度,分类器的设计不能过于复杂。可以想到的解决思路之一是减少要分类的子区域,这是 R-CNN 的一个改进策略,其采用了 selective search 方法来找到最有可能包含目标的子区域(Region Proposal)。
对于 CNN 分类器而言,滑动窗口是非常耗时的,幸运的是,结合卷积运算的特点,我们可以使用 CNN 实现更高效的滑动窗口方法。有一种全卷积的方法,用卷积层代替了全连接层。overfeat 算法的思路是,如果输入图片大小是 16x16,经过一系列卷积操作,提取了 2x2 的特征图,这个 2x2 的图上每个元素都是和原图是一一对应的,相当于在原图上做大小为 14x14 的窗口滑动,且步长为 2,共产生 4 个子区域,最终输出的通道数为 4,可以看成 4 个类别的预测概率值,这样一次 CNN 计算就可以实现窗口滑动的所有子区域的分类预测。由于图片的空间位置信息的不变性,尽管卷积过程中图片大小减少,但是位置对应关系还是保存的。这个思路也被 R-CNN 借鉴,从而诞生了 Fast R-CNN 算法。
上面尽管可以减少滑动窗口的计算量,但是只是针对一个固定大小与步长的窗口,这是远远不够的。Yolo 算法很好的解决了这个问题,它不再是窗口滑动了,而是直接将原始图片分割成互不重合的小方块,然后通过卷积最后生产这样大小的特征图,基于上面的分析,可以认为特征图的每个元素也是对应原始图片的一个小方块,然后用每个元素来可以预测那些中心点在该小方格内的目标,这就是 Yolo 算法的朴素思想。
# 2.2 YOLO
# 2.2.1 Unified Detection
整体来看,Yolo 算法采用一个单独的 CNN 模型实现 end-to-end 的目标检测,整个系统如图所示:首先将输入图片 resize 到 448x448,然后送入 CNN 网络,最后处理网络预测结果得到检测的目标。相比 R-CNN 算法,其是一个统一的框架,其速度更快,而且 Yolo 的训练过程也是 end-to-end 的。
具体来说,Yolo 的 CNN 网络将输入的图片分割成
- 当该边界框是背景时(即不包含目标),
。 - 当该边界框包含目标时,
。
边界框的准确度可以用预测框与实际框(ground truth)的 **IOU(intersection over union,交并比)** 来表征。因此置信度可以定义为
边界框的大小与位置可以用 4 个值来表征:
每个单元格需要预测
# 2.2.2 Design
Yolo 采用卷积网络来提取特征,然后使用全连接层来得到预测值。网络结构参考 GooLeNet 模型,包含 24 个卷积层和 2 个全连接层。对于卷积层,主要使用 1x1 卷积来做 channle reduction,然后紧跟 3x3 卷积。对于卷积层和全连接层,采用 Leaky ReLU 激活函数:
在训练之前,先在 ImageNet 上进行了预训练,其预训练的分类模型采用图中前 20 个卷积层,然后添加一个 average-pool 层和全连接层。预训练之后,在预训练得到的 20 层卷积层之上加上随机初始化的 4 个卷积层和 2 个全连接层。由于检测任务一般需要更高清的图片,所以将网络的输入从 224x224 增加到了 448x448。
由于每个单元格预测多个边界框。但是其对应类别只有一个。那么在训练时,如果该单元格内确实存在目标,那么只选择与 ground truth 的 IOU 较大的那个边界框来负责预测该目标,而其它边界框认为不存在目标。这样设置的一个结果将会使一个单元格对应的边界框更加专业化,其可以分别适用不同大小,不同高宽比的目标,从而提升模型性能。大家可能会想如果一个单元格内存在多个目标怎么办,其实这时候 Yolo 算法就只能选择其中一个来训练,这也是 Yolo 算法的缺点之一。要注意的一点时,对于不存在对应目标的边界框,其误差项就是只有置信度,左标项误差是没法计算的。而只有当一个单元格内确实存在目标时,才计算分类误差项,否则该项也是无法计算的。
# 2.2.3 NMS
NMS 即非极大值抑制算法(non maximum suppression, NMS),这个算法不单单是针对 Yolo 算法的,而是所有的检测算法中都会用到。NMS 算法主要解决的是一个目标被多次检测的问题,如人脸检测的案例中,人脸可能被多次检测,但是其实预测只希望最后仅仅输出其中一个较好的预测框,那么可以采用 NMS 算法来实现这样的效果:首先从所有的检测框中找到置信度较大的那个框,然后挨个计算其与剩余框的 IOU,如果其值大于一定阈值(重合度过高),那么就将该框剔除;然后对剩余的检测框重复上述过程,直到处理完所有的检测框。Yolo 预测过程也需要用到 NMS 算法。
对于 Yolo 的预测过程,首先不考虑 batch,认为只是预测一张输入图片。根据前面的分析,最终的网络输出是 7730,但是我们可以将其分割成三个部分:类别概率部分为
所有的准备数据已经得到了,对于第一种策略来得到检测框的结果,我认为这是最正常与自然的处理。首先,对于每个预测框根据类别置信度选取置信度较大的那个类别作为其预测标签,经过这层处理我们得到各个预测框的预测类别及对应的置信度值,其大小都是
对于 Yolo 算法,其采用了另外一个不同的处理思路,与上段所述方法其区别就是先使用 NMS,然后再确定各个 box 的类别。对于每个 boxes,首先将小于置信度阈值的值归 0,然后分类别地对置信度值采用 NMS,这里 NMS 处理结果不是剔除,而是将其置信度值归为 0。最后才是确定各个 box 的类别,当其置信度值不为 0 时才做出检测结果输出。我们需要慎重选择 NMS 算法,因为其对 Yolo 的性能是影响很大的。
# 3. Face recognition
该章主要介绍如何利用 YOLO 算法实现人脸检测的功能,主要为代码的解析和实验说明。
# 3.1 Dataset
YOLO 本身使用的是 VOC 的数据集,但是我们的实验中需要使用自定义数据集来训练 YOLO 网络的话,就需要在 VOC 数据集基础上进行改造,按照 VOC 数据集的结构和格式来构建所需的数据集。在本实验中使用的数据集是 CelebA 大规模名人人脸标注数据集,CelebA 数据集中的图片命名格式为统一的 000001.jpg—202599.jpg,BoundingBox 的信息在 list_bbox_celeba.txt 中保存。
# list_bbox_celeba.txt
line 1: The number of picture
line 2: information
line 3: < xxxxxx.jpg > < [x1] [y1] [width] [height] >
其中
YOLO 训练时需要:
指明训练图片绝对路径的train.txt文件和指明验证图片绝对路径的val.txt。
所有图片对应的label文本文件,在voc数据集中位于VOC2007/labels文件夹内。
训练数据配置文件voc.data
网络配置文件,这里用tiny-yolo.cfg
类别名列表文件voc.names
其中 1,2 是由我们自己生成赖江 celebA 的数据集转化成 VOC 的数据集,而后对目录结构进行调整。
# 3.2 Darknet
Darknet 是一个开源的神经网络框架,其使用 C 和 CUDA。
# 3.2.1 Installing Darknet
git clone https://github.com/pjreddie/darknet.git
cd darknet
make
执行上述代码安装,我们使用的是 ubuntu 18.04LTS 操作系统。
wget https://pjreddie.com/media/files/yolov3.weights
下载好提前训练好的参数,然后就可以运行 detector 了。
./darknet detect cfg/yolov3.cfg yolov3.weights data/horses.jpg
运行后会产生如下的输出:
layer filters size input output
0 conv 32 3 x 3 / 1 416 x 416 x 3 -> 416 x 416 x 32 0.299 BFLOPs
1 conv 64 3 x 3 / 2 416 x 416 x 32 -> 208 x 208 x 64 1.595 BFLOPs
...
104 conv 256 3 x 3 / 1 52 x 52 x 128 -> 52 x 52 x 256 1.595 BFLOPs
105 conv 255 1 x 1 / 1 52 x 52 x 256 -> 52 x 52 x 255 0.353 BFLOPs
106 yolo
Loading weights from yolov3.weights...Done!
data/dog.jpg: Predicted in 21.323896 seconds.
dog: 99%
truck: 92%
bicycle: 99%
输出的预测图片为 predictions.png:
# 3.2.2 Face recognition
该系统的设计步骤为:在上节下载好的 darknet
文件夹中进行一些文件和脚本的修改,以便于我们进行人脸识别。
创建数据集文件;
采集需要识别的头像,放入文件夹中;
标注人名为类别名,基于 labelImg,其中类别名称为人名,并使用脚本生成文件名称列表;
修改
script
文件夹下面的voc_label.py
文件,执行过后,会生成标注文件对应的 txt 文件;修改配置文件
./cfg/voc.data
中的内容classes= 4 train = /home/user/darknet/scripts/homework_train.txt //valid = /home/pjreddie/data/voc/homework_test.txt names = data/voc.names backup = /home/user/darknet/result
修改 ``./data/voc.names`,将四个类别名称(人名)写进,原来的都删除掉。;
修改网络参数的最后一层,在文件
tiny-yolo-voc.cfg
中,修改如下:. . . [convolutional] size=1 stride=1 pad=1 filters=45 activation=linear [region] anchors = 1.08,1.19, 3.42,4.41, 6.63,11.38, 9.42,5.11, 16.62,10.52 bias_match=1 classes=4 coords=4 num=5 softmax=1 jitter=.2 rescore=1 object_scale=5 noobject_scale=1 class_scale=1 coord_scale=1 absolute=1 thresh = .6 random=1
训练网络
./darknet detector train ./cfg/voc.data cfg/tiny-yolo-voc.cfg
在漫长的等待过后,可以在
./darknet/results
生成最后的参数数据,其中tiny-yolo-voc_final.weights
是最后的参数,其他的都是过程参数。进行测试
./darknet detector test cfg/voc.data cfg/tiny-yolo-voc.cfg results/tiny-yolo-voc_final.weights data/images.jpg
至此,整个过程完成。