基于双流卷积神经网络的人体行为识别方法①

刘 云,张 堃,王传旭

(青岛科技大学 信息科学技术学院,青岛 266000)

1 引言

随着各种摄像监控设备的快速发展,视频和图像的数据量在不断增加.如何分析视频图像中的信息也成为一个热门的研究内容,视频分析中的一个重要分支就是行为识别.人体行为识别的目标是从一个未知的视频或者是图像序列中自动分析其中正在进行的行为,目前对于行为识别的研究热点主要是对短视频中单个行为的识别,而在实际生活及应用中,更多的视频数据是包含多个不同行为的复杂长视频.这就需要使用另一种识别算法:时序行为检测(temporal action localization).这种算法任务要求检测出长视频中每个行为的类别,同时要标注出每个行为的开始时间和结束时间.这种算法可以应用到许多方面,比如自动检索和智能监控等.

时序行为检测通常可以分为两个阶段,提议生成阶段和分类识别阶段.提议生成阶段的主要目标是生成可能含有行为动作的视频片段,视频片段称为行为提议,而分类识别阶段的任务则是对提议生成阶段产生的行为提议进行识别分类,并且进一步确定行为类别和起止时间.尽管目前传统的行为识别已经达到较高的准确度,但是在确定行为起止时间上仍然不尽如人意[1,2].因此,如何产生高质量的行为提议,成为该内容的一个重点研究方向[3-6].为了获得高质量的提议,提议生成阶段产生的提议在持续时间上需要灵活可变,用于应对视频片段持续时间长短不一并且差距较大的问题,同时产生的提议应具有精确的时间边界.最近的一些提议生成方法[3-5,7]利用不同长度的滑动窗口来生成提议,然后使用训练好的模型来评估提议的置信度,但是,这种预先定义持续时间和间隔时间来产生提议的方法有一些明显的缺点:(1) 起止时间的精确度不足；(2) 固定的行为片段长度无法处理不同持续时间的行为动作,而在不同行为动作持续时间差距较大时,更会出现无法满足不同持续时间的要求,而增多滑动窗口的数量又会带来大量冗余的计算.

最近的研究[7-9]将深度神经网络应用到检测框架中并且获得了较好的性能表现.S-CNN[7]提出了一个多阶段的卷积神经网络,该算法通过使用定位网络提高了识别精度.然而,S-CNN 使用滑动窗口产生行为提议,C3D[10]作为特征提取器最初用于单元分类器,只能容纳16 帧作为输入,在应对时序行为检测任务时,需要消耗大量的时间进行计算.另一项研究[8]使用递归神经网络(RNN)来学习预测动作的起点和终点的一种策略.这种顺序预测对于处理长视频通常非常耗时,并且它不支持用于特征提取的逐帧CNN 的联合训练.

本文在上述背景下,为了克服滑动窗口的缺点,生成高质量的行为提议,本文提出了基于双流卷积神经网络[11]的时序行为检测模型.该模型基于双流卷积神经网络提取的特征,产生覆盖时间灵活可变的行为提议,之后送入多层感知机中进行边界迭代回归,然后将行为提议扩展为三段式的特征序列设计,最后输入分类器中进行动作分类.

2 识别模型

本文提出一种基于双流卷积神经网络的模型,如图1所示.首先使用双流卷积神经网络提取长视频的特征序列,然后将该特征序列作为模型的输入,使用Temporal Actionness Grouping (TAG)[12]方法在特征序列上灵活地生成行为提议.利用多层感知机对每一个行为提议的起止边界进行迭代操作,这一过程可以更为精细地处理行为提议的边界,使之更加贴近真实的边界信息.每一个行为提议都会使用三段式特征描述重新设计,三段式设计将行为提议划分为开始区间、进行区间和结束区间,按照前后顺序对应拼接相应的特征序列.最后对包含目标动作的行为提议进行行为识别,获得分类结果.

图1 基于双流卷积神经网络的人体行为识别模型

2.1 问题描述

一个未分割的长视频可以表示为X=其中xn表示视频X中的第n帧.视频X的动作标注由一组动作实例组成,An是视频X中真实动作实例的数量,ts,n,te,n分 别是动作实例 φn的开始时间和结束时间.本文算法的任务就是自动定位每段行为的起止位置并识别它们的行为属性.

2.2 特征序列提取

为了提取双流卷积神经网络特征,将视频划分为T个连续等长且无重叠的单元,则视频可以表示为T表示视频中单元的数量,一个单元st=xtn,otn表示两部分的内容,xtn是视频X中的第tn个RGB 帧,otn是 以xtn帧为中心,附近的堆叠光流场.为了减少计算损耗,使用规律的帧间隔提取单元.本任务所用数据集中的视频数据量大,相邻的帧信息冗余度较高,密集采样耗时且不必要,因此使用规律的帧间隔提取单元,在每个单元上获取特征,可以在保证信息完整度的前提下降低计算损耗.

给定一个单元st,在空间和时间网络的顶层连接输出分数以形成编码特征向量ftn=(fS,tn,fT,tn) ,其中fS,tn,fT,tn分别表示空间网络和时间网络的输出向量.因此给定一个长度为ls的单元序列S,可以提取出特征序列双流卷积特征序列将被送入TAG 网络中生成行为提议.

2.3 行为提议

相比较于滑动窗口而言,TAG 方法能灵活的生成不同长度的动作提议,同时并不需要大量的计算.TAG 方法使用了一个行为分类器来评估每个单元中发生动作的概率,这个行为分类器是一个二元分类器.该方法的基本思想是找到高动作概率的连续区域,为了实现这个目的,该方法重新设计了一个经典的分水岭算法,并把它应用到了一维的动作概率值上.该方法通过设置不同的“水位”可以得到一系列的“盆地”,每一个盆地对应了时域范围内一段高动作概率区域.

给定一系列的盆地G,选用了一种类似于文献[13]的聚类方法,这种方法试着连接小盆地变成行为提议区域.该方案的工作流程如下:先从一个种子盆地开始,并且连续吸收随后的盆地,直到盆地部分在整个持续时间内(即从第一个盆地开始到最后一个盆地结束)的部分下降到某个阈值Y以下.通过这种方法,可以从不同的种子盆地开始产生一组区域,用G′(τ,γ)来表示.注意 τ 和 γ并不是选择好的特定组合,而是均匀地从(0,1)之间采样,步长为0.05.这两个阈值的组合将会产生多组区域.然后,将他们结合起来,并使用非极大值抑制的方法过滤重叠度高的区域,设置IoU 阈值为0.95.生成的行为提议将被送入多层感知机中边界回归.

2.4 边界回归

时域上进行边界回归的基本思路是利用神经网络推断行为提议的边界.本文使用多层感知机作为回归网络,将行为提议作为输入,输出坐标回归偏移量,具体计算如式(1).

其中,sclip,eclip分别是输入的行为提议的开始和结束坐标,sgt,egt分别是与之对应的真实数据的开始和结束坐标.本文使用的坐标回归模型有两个优点:第一,使用单元级坐标回归,这与双流卷积神经网络基于单元提取特征的方式相匹配,计算消耗也比较小；第二,不使用坐标参数化,直接使用起始坐标的偏移量作为回归结果.这是因为行为提议的坐标回归在时域进行,而空间坐标回归在空间域进行,由于相机投影,目标可以在图像中重新缩放,因此需要先将边框坐标标准化为某个标准尺度.而时域坐标可以依靠时域本身作为标准尺度,不需要进行参数化.

在训练边界回归网络时,需要给行为提议分配标签用以判断该行为提议中是否包含行为.对于一个行为提议,计算它和所有标定好的真实数据的tIoU(temporal Intersection over Union)重叠值,如果其中的最大值超过了0.5,则将最大值对应的真实数据的边界和类别信息赋予该行为提议.并将该行为提议视为正样本,即含有行为,否则视为负样本.

如图2所示,本文的边界回归任务由多层感知机使用迭代的方式完成,边界回归的输出结果作为输入再次送入多层感知机中进行计算,重复多次以获得更为精确的结果.该回归模型将行为提议作为输入,输出时域上的坐标回归偏移量,计算之后得到回归后的边界坐标值.对于该层网络,给定一个候选提议的边界数据输入值pc=[ts,te],输出数据p1c=[t1s,t1e]会作为输入进行第二轮的边界回归计算,第二轮的输出为p2c=[t2s,te2].迭代过程总共进行K次,最后的边界结果为:

2.5 提议特征

为了建立如图3所示的提议特征 φ,对于一个行为提议,将提议本身的范围定义为进行区间pc=[ts,te],提议 φ的持续时间为d=te-ts.与它相关的开始区间和结束区间分别为ps=[ts-d/4,ts+d/4]和pe=[te-d/4,te+d/4].对应选择开始、结束和进行区间三部分对应的特征序列,将这些向量前后拼接,即可获得候选提议 φ的提议特征fφ=(fps,fpc,fpe).该提议特征具有很好的鲁棒性,在引入开始区间和结束区间后,使得行为提议特征 具备了上下文信息.

图2 边界回归网络处理行为提议边界

图3 行为提议特征构建

2.6 行为分类

深度学习网络常用的分类器,本文选择使用多层感知机网络作为特征构建后的多分类器.对于时序行为检测任务,多层感知机网络输出n+1 个概率值,其中n表示数据集中行为的数量,1 表示背景类.在ActivityNet v1.3 中,n=200,在THUMOS 2014 中,n=20.每个概率值表示属于某一类行为的概率,将最大概率值对应的行为作为行为分类的结果.

为了获取较好的实验结果,本文使用一个多任务损失函数来联合训练边界回归和行为分类网络.损失函数如式(3)所示.时序行为检测任务需要对行为定位和识别,这两个任务息息相关,如果单独训练网络会降低识别的泛化能力,可能会出现对某一任务的过拟合现象.而联合训练可以较好的解决这个问题,联合训练可以在有限的数据集内完成训练,由于引入了额外的相关训练数据,有助于网络学习到更适合任务需求的参数,可以提高模型的泛化能力.行为的类别和发生时间是个体属性的不同方面,具有较强的相关性,使用联合训练可以使得定位与识别任务真正地结合起来,学习到的内容彼此受益,提高时序行为检测的准确率.

其中,Lcls是分类损失函数,对于本文中多分类任务而言,使用多分类交叉熵函数作为损失函数.Lreg是边界回归损失函数,λ是超参数.回归损失函数为:

其中,R是曼哈顿距离,N是batch size,n是行为类别的总数量,是标签,当第i个样本属于z类时,=1,否则,lzi=0.o′是回归偏移量,o是真实数据.学习率设置为0.005,batch size 设置为128.

3 实验

为了验证本文算法的有效性,本文在ActivityNet v1.3[1]和THUMOS 2014[2]数据集上进行实验.ActivityNet v1.3 数据集是常用的时序行为检测数据集,包括200 类不同的动作,同时提供了边界和种类信息标注.THUMOS 2014 中没有训练集,有20 类行为带有标注.本文分别在两个数据集上进行实验,在各自提供的数据子集上训练网络,并使用预训练的网络进行测试,将实验结果与现有方法进行对比分析.

3.1 数据集

ActivityNet v1.3[1]是一个用于时序行为检测的大型数据集,其中包含19994 个带有200 类动作标注的长视频,在2017年和2018年的ActivityNet 挑战中使用了该数据集.ActivityNet 按照2:1:1 的比例分为训练集、验证集和测试集.

THUMOS 2014[2]有1010 个视频用于验证,1574个视频用于测试.这些视频中包含20 类带有行为标注的目标动作.该数据集没有训练集,使用UCF101数据集作为训练集.由于训练集没有提供时间注释,本文在验证集上训练模型并在测试集上进行实验测试.因此将带有20 类行为标注的220 个视频用于训练.在本文的实验中,将本文提出的方法与THUMOS 2014和ActivityNet v1.3 上的现有技术进行比较,并进行结果分析.

3.2 实验网络参数设置

本文实验环境选择深度学习框架Caffe 平台实现.使用SGD 方法学习模型中的参数,batch size 为128,momentum 为0.9.双流卷积神经网络采用ResNet 网络用作空间网络,BN-Inception 网络用作时间网络.空间网络和时间网络的初始学习率分别设置为0.001 和0.005.在ActivityNet v1.3 中,空间网络和时间网络迭代训练次数分别为9500 次和20 000 次,学习率分别在迭代每4000 次和1000 次后缩小0.1.在THUMOS 2014 中,空间网络和时间网络分别进行1000 次和6000 次的迭代训练,学习率在每400 和2500 次时缩小0.1.在特征提取过程中,单元间隔均被设置为16.在TAG 方法中使用的二元行为分类器使用每个数据集的训练集进行训练.在边界回归过程中,K=3.

3.3 实验结果分析

评价标准:AvtivityNet v1.3[1]和THUMOS 2014[2]都有统一的评价标准,因此按照它们的评价标准测试不同IoU 阈值的平均预测精度mAP.在ActivityNet v1.3 数据集中,所需测试的IoU 阈值为{0.5,0.75,0.95},IoU 阈值范围[0.5:0.05:0.95]的mAP 的平均值用于比较不同方法之间的性能.在THUMOS 2014 数据集中,所需测试的IoU 阈值为{0.1,0.2,0.3,0.4,0.5}.阈值为0.5 时得出的平均预测精度用于比较不同方法的实验结果.

将本文算法与其它时序行为检测方法在THUMOS 2014 数据集和ActivityNet v1.3 数据集上进行比较,如表1、表2所示.从表1、表2中可以发现,在这两个数据集上,本文提出的算法识别准确率优于其它算法,识别效果较好.本文使用双流卷积神经网络所获取的特征结合了运动表层特征和时序信息两部分,更好的发掘了视频所包含的信息.行为提议在经过多层感知机迭代处理后边界信息更为准确,之后的三段式特征设计融合了上下文信息,一方面建立了较为全面的行为描述,另一方面提高了行为识别准确率.

4 结论与展望

为了充分获取视频中的时空信息,使用双流卷积神经网络构建特征描述符,之后通过TAG 方法产生候选行为提议,经过多次迭代处理后获取更为准确的边界信息,将行为提议扩展为三段式特征设计,并对目标行为进行识别.该方法在结合时序信息的基础上,生成了质量较高的动作提名,时序边界更为准确,识别率也有所提升.实验结果表明该方法能在THUMOS 2014 数据集合ActivityNet v1.3 数据集上得到较好的效果.但是行为提议生成和回归的方法着眼于局部信息,缺少与行为提议全局特征的结合分析,时序定位的准确度仍有不足.下一步的研究将会引入行为提议的特征共同分析定位准确度,获得更为准确的时序边界.

表1 不同时序行为检测算法在THUMOS 2014 数据集上 的准确率(%)

表2 不同时序行为检测算法在ActivityNet v1.3 数据集上的准确率(%)