APP下载

一种基于多尺度特征融合的人头计数检测方法研究

2021-06-18范自柱石林瑞符进武

华东交通大学学报 2021年2期
关键词:人头集上尺度

张 泓,范自柱,石林瑞,符进武

(华东交通大学理学院,江西 南昌330013)

人群计数一直是计算机视觉领域的热门问题,有许多重要的应用,如公共安全管理,灾难管理,公共空间设计,情报收集及分析和嫌疑人搜索[1],这些各式各样的应用促使研究人员去开发各种适用于不同环境的人群计数方法。目前的人群计数方法主要有两种:①基于回归的方法;②基于检测的方法。

在基于回归的方法中,研究人员[2-3]将人群整体当作一个对象来进行人群计数,Wang[2]提出了一个端到端的卷积神经网络模型来对密集场景中的人群进行计数,他们的模型直接输出密集人群图像中的人数。陆金刚[3]设计了一个基于多尺度多列卷积神经网络(multi-scale multi-column convolutional neural network,MSMCNN),每一列网络使用不同大小的卷积核来提取不同尺度的特征以便感受到不同尺度的人头。但是这些方法都将人群整体作为研究对象,直接输出场景里的人数或与输入图像对应的密度图,无法精确的定位到场景中的每一个人。

在基于检测的方法中,研究人员[4-8]使用基于头部检测的方法来对场景中的人群进行计数。李欢[4]提出了一种基于SSD的行人头部检测方法,添加类别预测和位置预测两个旁支网络实现特征分离来改进SSD。张晓琪[5]提出了一种基于多特征协同的人头检测方法,相对于传统方法检测方法精度上更好。邢志祥[6]对经典的目标检测方法进行了实验,通过选取不同的特征提取网络和检测网络进行搭配,得到了一种最优的检测方法。Vora[7]设计了一个轻量级的人头检测网络,他们使用VGG16作为特征提取网络,然后将最后一层特征送入一个3*3大小的卷积核,最后分别使用两个1*1大小的卷积做分类和回归。Merad[8]提出了一种结合跟踪过程的头部检测方法,他们通过头部参考系统与摄像机参考系统之间的刚体变换来估计头部姿态。但是这些方法在高遮挡,尺度差异大的场景中精度很低。

本文提出了一种基于改进的Faster-RCNN[9]的头部检测方法,该方法可以有效的感知多尺度信息,具有较高的准确性和鲁棒性。

1 基于多尺度特征融合的Faster-RCNN网络

事实上,在一张静态图片中,靠近摄像机的人头往往较大,远离相机的人头较小,使用单一的特征图进行检测往往不能取得很好的效果。为了解决这个问题,本文提出了一个新的人头检测方法,该方法能有效的利用多尺度特征来检测密集场景中的人群[10]。网络结构如图1所示,该的方法基于Faster-RCNN,为了进一步提高精度,设计了一个多尺度特征融合模块 (multi-scale feature fusion,MSFF),该模块可以融合不同层的特征,加强特征图之间的相关性。此外,通过先验框的设计和使用Roi-Align层代替了原来网络中的Roi-Pooling层,提高了在IoU(intersection over union)等于0.7时的检测效果。为了进一步提高检测效果使用大分辨率的图片进行训练,在保证图片的宽高比不变的情况下,将输入的图片放大到1 024*768,送入到特征提取网络中提取特征,然后将得到的特征送入多尺度特征融合模块进行融合,将融合后的特征图送入到RPN(region proposal network)网络中提取候选框,这些不同大小的候选框会通过Roi-Align映射成相同大小的特征图,最后将这些特征图送入到后续的网络中进行分类和回归。

图1 基于多尺度特征融合的Faster-RCNNFig.1 Faster-RCNN based on multi-scale feature fusion

1.1 先验框的设计

通过对数据集中人头标注框的分析发现,人头的包围框可以近似看成一个正方形。去除了原Faster-RCNN网络中宽高比为0.5和2先验框,只保留了宽高比为1的先验框,即在每个特征图的每个像素点上只生成1种尺度的候选框。如图2所示,多尺度特征融合模块得到的N2-N6特征图分别对应的感受野(特征图上的像素点映射到原图所对应像素区域)的大小为{32^2,64^2,128^2,256^2,512^2}(32^2表示32*32大小的像素区域)。

1.2 多尺度特征融合模块

在一个深度特征提取网络当中,深层的特征包含着丰富的语义信息但不利于小物体的检测,浅层的特征包含着丰富的位置信息但语义信息则较少。若只使用最后一层特征来进行检测,其它层的特征信息就会被忽略。Lin等[11]设计了一种具有横向连接的自顶向下的特征金字塔网络结构(feature pyramid networks,FPN),该结构可以融合特征提取网络中不同尺度的特征来提升物体检测的准确率。我们参考了Liu等[12]在FPN结构上的改进,增加了一条自底向上路径,如图2所示,C1-C5表示大小为1 024*768的输入图片经过ResNet101的第一到第五层卷积等得到的特征图,P5由C5经过一个卷积核大小为1*1的卷积层得到,P4由横向连接的C4和下采样的C5得到,P3-P2以此类推,P2-P5的通道数都为512,特征图的大小分别为(256*192,128*96,64*48,32*24)。N2由横向连接的P2得到,N3-N5得到的方式与P2-P3类似,N6由N5上采样得到。此外,为了降低上采样带来的混叠效应,在N3-N5之后分别加上了一个3*3的卷积层,但保持特征图的大小不变,N2-N6的通道数都为256,特征图的大小分别为(256*192,128*96,64*48,32*24)。这样做的目的是充分利用特征融合,使用五层不同大小的特征图分别进行预测,能够极大的提高网络对不同尺度人头的检测效果。

1.3 Roi-Align

图2 多尺度特征融合模块Fig.2 Multi-scale feature fusion module

此外,受Mask-RCNN[13]的启发,将Faster-RCNN里的Roi-Pooling换成了Roi-Align。为了使检测网络能够输入任意大小的图片,Roi-Pooling被用来将感兴趣区域池化成固定大小的特征图,以便后续网络进行分类和回归的操作。但Roi-Pooling有一个局限性,Roi-Pooling进行池化操作的时候会有两次量化过程,即对浮点数进行向下取整。这样一来,经过Roi-Pooling得到的候选框和最开始候选框会有一些偏差,这在IoU较低的情况下影响不大,但IoU较高时精度会有很大的影响,特别是对于小物体。假设原图中有一个区域的大小为655*655,经过ResNet101特征提取网络后在最后一层特征图的大小为20.47*20.47,这个时候Roi-Pooling会进行向下取整操作,最后得到的特征图的大小为20*20。将20*20大小的特征图映射成大小为7*7大小的特征图时,Roi-Pooling平均将20*20的特征图划分成7*7的大小,也就是49个小区域,每个区域的大小为2.86*2.86,这个时候Roi-Pooling又会进行第二次量化操作,小区域大小变成2*2的大小。Roi-Align对Roi-Pooling进行了改进,Roi-Align取消了量化操作,使用双线性插值法来计算最后的值,可以得到更精确的候选框。

2 实验

实验代码基于Python3.5,使用的深度学习框架为Pytorch(0.4版本),显卡为RTX 1080ti,内存为16 G,运行环境Ubuntu16.04。

2.1 实验数据集

本文分别在HollwoodHeads[14],Brainwash[15]这两个公开的数据集上测试了该模型。如表1所示,Brainwash数据集是一个密集人头检测数据集,拍摄的是在一个咖啡馆里出现的人群,然后对这群人进行标注而得到的数据集。该数据集包含3个部分:训练集由10 769张图像组成,共包含81 975个人头;验证集由500张图像组成,共包含3 318个人头;测试集由500张图像,共包含5 007个人头,平均每副图像包含7.64个人头。HollwoodHeads数据集是在21部电影场景里收集的,由224 740张图像构成,共369 846个人头,平均每张图片包含1.65个人头。

表1 数据集介绍Tab.1 Introduction to dataset

2.2 超参数设置

训练过程中的初始学习率为0.001,训练16万次,训练到80 000次时进行一次学习率的调整,调整大小为初始学习率的十分之一。使用随机梯度下降法(stochastic gradient descent,SGD)来优化模型,动量(momentum)的大小设置为0.9。使用非极大抑制(non-maximum suppression,NMS)来去除重复比例较大的候选框。正负样本的判定与Faster-RCNN的设置一样,小于0.3的样本作为负样本,大于0.7的样本作为正样本,在此之间的样本忽略不计。使用VOC[16]的评价标准来计算平均精度(AP),在测试时,对于输出的检测框如果与实际框的IoU比值大于预先设定的阈值(0.5和0.7),那么就认为这个检测框检测到了人头。

2.3 实验对比

对于Brainwash数据集,为了方便训练,去除了数据集中的没有人头的图片,构成的新数据集包含10 461张训练图片,484张测试图片,493张验证集图片。表2给出了该模型与之前一些经典方法的对比,Sermanet等[17]提出了一种多尺度和滑动窗口的方法,他们称之为Overfeat-AlexNet模型。Stewart[15]用该方法进行了密集场景下的人头检测实验,并进一步提出了一个基于LSTM的人头计数方法,通过GoogLeNet来提取图片中的特征,然后将这些特征送入到一个LSTM中进行解码,输出一系列的人头检测框。通过使用不同的损失函数,他们提出了3种不同的检测模型,分别是:ReInspect,Lfix;ReInspect,Lfirstk;ReInspect,Lhungarian。除此之外,我们还与其他的一些经典的检测方法做了对比SSD[18],YOLO9000[19],Tiny[20]和HeadNet[21]。如表2所示,本文方法在Brainwash数据集上取得了最高的精度。图3展示了本文方法和FCHD模型在Brainwash数据集上的检测结果,从图3(a)中可以看到FCHD模型对一些小人头很难进行有效的检测。图3(b)展示了本文方法的检测结果,可以看到对一些小人头本文方法也能检测出来(其中黄色框是检测框,蓝色框是真实标注的人头)。值得一提的是,数据集中有一些没有标注的人头,如图3(b)中红色箭头所指,这些数据或许会对精度的计算产生不利的影响。

表2 在Brainwash上的对比实验Tab.2 The comparison experiment on Brainwash

图3 模型检测效果对比Fig.3 Model detection result comparison

HollwoodHeads数据集是一个电影场景里收集的数据集,训练集,验证集,测试集分别包含216 694,6 719,1 297张图片。我们和一些基线方法进行了对比,如SSD[18],YOLO9000[19],Tiny[20],HeadNet[21]的人头检测方法,Merad[8]提出的两种人头检测模型Context-Local和Context-Local+Global+Pairwise方法。表3展示了本文方法在IoU=0.5和0.7的情况下都优于以前的方法。图5展示了一些本文方法的检测结果,对于电影场景中正常的人头都能有一个很好的检测,但是对于一些影子数据,该方法会有漏检测的结果。

表3 在HollwoodHeads上的对比实验Tab.3 The comparison experiment on Hollwood Heads

图4 HollwoodHeads数据集检测示例Fig.4 Detection result on Hollwood Heads

2.4 进一步的实验

为了验证多尺度融合模块和Roi-Align的有效性,在Brainwash数据集和HollwoodHeads数据集上进行了一些实验。IoU的大小表示检测框和真实框之间的重合度,IoU越大则表明检测框和真实框之间越接近。一般IoU大于等于0.5就可以认为是一个正确的检测,但在某些场景中对检测框有更高的要求,本文在IoU=0.5和0.7的这两种情况下进行了实验。表4展示了在Brainwash数据集上的实验结果,可以看到本文方法比原始的Faster-RCNN网络提高了15.7%,将RoI_Pooling替换为Roi_Align在IoU=0.5时候精度相差不大,而在IoU=0.7的情况下可以带来6%的提升。这说明Roi_Align使得模型输出得框更接近真实得框。

表4 在Brainwash数据集上的实验结果Tab.4 Experimental results on Brainwash

表5展示了在HollwoodHeads数据集上的实验结果,Faster-RCNN(C4)表示用特征提取网络第4层特征(C4)进行的实验。在HollwoodHeads数据集上本文方法比原始的Faster-RCNN网络提高了8.28%,将RoI_Pooling替换为Roi_Align在IoU=0.7的情况下可以在此基础上再带来1.4%的提升。

表5 在HollwoodHeads数据集上的实验结果Tab.5 Experimental results on Hollwood Heads

此外,我们还在不同分辨率下的训练图片进行了实验,实验结果如表6所示,在保持图片宽高比不变的情况下1 024*768表示将图片放大到1 024*768的大小,640*480相同,IoU的取值为0.5,由表6可知在相同条件下,输入图片分辨率越大精度越高。

表6 在不同分辨率下的实验结果Tab.6 Experimental results at different resolutions

3 结论

本文提出了一种端到端的人头检测模型,该模型能够很好的检测到密集场景的人头。

1)使用ResNet101作为特征提取网络提取特征,然后将提取到的特征送入一个多尺度特征融合模块(MSFF)进行特征融合,该模块将融合后的特征分别送入后续的网络进行检测。

2)通过先验框的设计和使用Roi-Align代替了Roi-Pooling进一步的提高了精度。由于融合了不同层的特征,该模型能够有效的检测到不同尺度的人头。实验表明,该方法在Brainwash和Hollwood-Heads数据集上达到了最优的结果。

猜你喜欢

人头集上尺度
基于双空间模糊邻域相似关系的多标记特征选择
环境史衰败论叙事的正误及其评判尺度
关于短文本匹配的泛化性和迁移性的研究分析
假人头防盗大法
“猎头”行动
朋友圈
师如明灯,清凉温润
以长时间尺度看世界
9
几道导数题引发的解题思考