基于多流深度学习的人体行为识别方法与流程

文档序号:21038839发布日期:2020-06-09 20:33阅读:449来源:国知局
基于多流深度学习的人体行为识别方法与流程

本发明涉及人体行为识别技术领域,尤其是涉及一种基于多流深度学习的人体行为识别方法。



背景技术:

近年来,人体行为识别作为视频分析的主要研究方向,已经广泛应用在视频游戏、机器人、人机交互、医疗保健等领域中。人体行为识别的过程主要指通过对视频中能够代表人体行为的特征提取,确定人体行为模式。现有的人体行为识别方法主要包括传统的行为识别方法,即基于人工设定的特征提取方法进行行为识别和基于深度学习框架的行为识别。传统的行为识别方法难以满足人体行为识别研究中空间复杂性高、时间差异性大的特点,不能实现高效的行为识别目的。而基于深度学习模型的人体行为识别方法可以通过训练从大规模视频数据中获得更加丰富的知识,提高对复杂人体行为的识别能力。目前人体行为识别的存在的困难主要有以下几个方面:严重依赖物体和场景,光流的长度和语义问题,卷积神经网络本身存在的“黑盒”问题等。而就人体所在的具体环境而言,受到的挑战主要包括遮挡、光照变化、自由度较大等。这些问题既是人体行为识别领域严重中存在的客观现象,也是人体行为识别研究必须解决的问题。

深度学习方法主要是通过大量数据集的训练得到深度学习模型,并在测试中不断完善,能够更有效扑捉到数据的波动特点。目前,主流的基于深度学习的人体行为识别方法分为基于3维卷积神经网络(c3d)的方法法和双流法。c3d的最大优势在于速度,通过3d卷积在空间和时间维度上捕捉视频流的运动信息,构造3d卷积网络进行人体行为识别。carreiraj等人提出了i3d模型(inception-v13d),基于inception-v1模型,将2d卷积扩展到3d卷积。dibaa等人提出了t3d模型(temporal3dconvnets),一方面是采用了3ddensenet,区别于之前的inception和resnet结构;另一方面,ttl层,即使用不同尺度的卷积(inception思想)来捕捉讯息。qiuz等人改进resnet内部连接中的卷积形式,利用提出的p3d模型(pseudo-3dresidualnetworks)进行行人识别。dibaa等人则设计了一个通用的模型len(linearencodingnetworks),用于网络提取出特征之后的处理,可以用于two-stream以及c3d中。

双流法对视频序列中每两帧计算密集光流,得到密集光流的序列,对视频图像和密集光流融合分析以完成识别任务。相比c3d法,双流法在主流公共数据集(如ucf101、hmdb51)中取得了更加突出的性能;由此,也衍生出了更多基于双流法的人体行为识别方法,例如双流融合人体行为识别方法和基于时空网络的人体行为识别方法(tsn)方法。此外,还有其他的人体行为识别方法涌现出来,例如将时空轨迹信息与卷积神经网络结合的方法,将三维特征与时空特征结合的方法等。

基于双流法的行为识别方法,通常包含多种模态数据,研究人员经常尝试不同的输入模式。因此,哪种模式的网络流在行为识别中表现更为重要,是基于双流法的行为识别方法研究的关键。在行为识别的研究中,现有的输入模式主要有rgb信息,光流信息,以及扭曲光流信息。而这三种信息都输入全局信息。rgb信息作为一个输入模式,是获取人体行为全局表征特征的关键。文献[sevillalaral,liaoy,guneyf,etal.ontheintegrationofopticalflowandactionrecognition[j],2017.]认为光流场由于其固有的尺度不变特性,对行为识别性能有较大的帮助。这也证明了光流特征作为一种模式输入在行为识别中的重要性。虽然近年来对motion信息的关注逐渐上升,指责行为识别过度依赖背景和外貌特征,而缺少对运动本身的建模。时序卷积网络(temporalsegmentnetworks,tsn)模型和基于注意的时间加权cnn(atw)的方法,都将扭曲光流特征作为一种模式输入,前者提出使用多个双流网络,分别捕捉不同时序位置的短时信息,然后进行融合,得到最后结果,目的是为了解决长时处理的问题。但在deepmind的研究中,当视频数据集的大小很大时,深度学习网络可以得到比仅使用光流更好的结果,这主要是因为当数据集足够大时,它可以覆盖复杂的光照、纹理和背景。因此,扭曲光流并非关键模式。

综上所述,现有基于双流法的行为识别方法虽然效果明显,但对于不同模态的重要性并没有分析,而且都采用视频全局信息,忽略了局部信息在视频分析中的重要性,局部信息缺失造成的特征提取不完备,同样会导致识别精度急剧下降。因此,需要研究一种新的人体行为识别算法。



技术实现要素:

有鉴于此,本发明的目的是针对现有技术中的不足,提供一种基于多流深度学习的人体行为识别方法,结合了全局空域信息、时域信息和局部空域信息,一方面,融合了局部时空信息,将rgb图像分成三个部分,用于提取视频帧局部的细节;另一方面,实现了多种模态融合,提高了识别结果的准确性。

为达到上述目的,本发明采用以下技术方案:

基于多流深度学习的人体行为识别方法,包括以下步骤:

s1、分别建立基于bn-inception网络的全局空域信息网络训练模型、局部空域信息网络训练模型和时域信息网络训练模型;

s2、训练并优化全局空域信息网络训练模型、局部空域信息网络训练模型和时域信息网络训练模型的参数,获取全局空域信息网络模型、局部空域信息网络模型和时域信息网络模型;

s3、将待识别视频v划分为s段非重叠的视频片段;

s4、在每个视频片段中随机提取一帧rgb图像以及rgb图像的分块图像和光流场特征,rgb图像作为对应视频片段的全局空域信息输入全局空域信息网络模型,分块图像作为对应视频片段的局部空域信息输入局部空域信息网络模型,光流场特征作为对应视频片段的时域信息输入时域信息网络模型;

s5、全局空域信息网络模型输出rgb图像的类别分数,局部空域信息网络模型输出分块图像的类别分数,时域信息网络模型输出光流场特征的类别分数;

s6、采用段共识函数对s段视频片段中的rgb图像的类别分数进行融合产生rgb图像段共识,采用段共识函数对s段视频片段中的分块图像进行融合产生分块图像段共识,采用段共识函数对s段视频片段中的光流场特征的类别分数进行融合产生光流场特征段共识;

s7、采用段共识函数对s段视频片段中的rgb图像段共识、分块图像段共识和光流场特征段共识进行加权融合,输出待识别视频的人体行为分类得分。

进一步地,在步骤s2中,采用数据集imagenet分别对全局空域信息网络训练模型、局部空域信息网络训练模型和时域信息网络训练模型进行训练。

进一步地,在步骤s2中,采用数据集ucf-101和数据集hmgb51分别对训练后的全局空域信息网络模型、局部空域信息网络模型和时域信息网络模型进行参数优化。

进一步地,对于待识别视频v,把它按相等间隔分为s段,为{v1,v2,…,vs},待识别视频v的动作识别公式为:c=h(f(g(rand(v1),w),…,g(rand(vs),w)));

其中,f表示段共识函数,c表示类别分数,g表示在训练阶段使用的对应类别的真值,h采用softmax预测函数,对待识别视频中人体行为分类得分进行预测;

最终的损失函数为:

其中,y表示训练数据集中的标签,f表示预测的结果,l(y,f)表示训练过程中使用的损失函数。

进一步地,所述分块图像的分块方式为:采用由上到下水平平均分配的方法将每一帧rgb图像分为大小相等的三个部分,然后分别将这三个部分的图像信息得出的类别分数加权融合得到分块图像的类别分数。

进一步地,在步骤s2中,全局空域信息网络模型、局部空域信息网络模型和时域信息网络模型均采用单独训练。

进一步地,在步骤s2中,rgb图像、分块图像和光流场特征的输入模式的像素大小均调整为32×32。

进一步地,在步骤s4中,全局空域信息网络模型和局部空域信息网络模型共享参数。

本发明的有益效果是:

本发明针对现有技术中的行为识别方法采用视频全局信息,而忽略了局部信息在视频分析中的重要性,局部信息缺失造成的特征提取不完备,同样会导致识别精度急剧下降,提供一种基于多流深度学习的人体行为识别方法,该识别算法包括以下步骤:首先,分别建立基于bn-inception网络的全局空域信息网络训练模型、局部空域信息网络训练模型和时域信息网络训练模型;之后,训练并优化全局空域信息网络训练模型、局部空域信息网络训练模型和时域信息网络训练模型的参数,获取全局空域信息网络模型、局部空域信息网络模型和时域信息网络模型;之后,将待识别视频v划分为s段非重叠的视频片段;之后,在每个视频片段中随机提取一帧rgb图像以及rgb图像的分块图像和光流场特征,rgb图像作为对应视频片段的全局空域信息输入全局空域信息网络模型,分块图像作为对应视频片段的局部空域信息输入局部空域信息网络模型,光流场特征作为对应视频片段的时域信息输入时域信息网络模型;之后,全局空域信息网络模型输出rgb图像的类别分数,局部空域信息网络模型输出分块图像的类别分数,时域信息网络模型输出光流场特征的类别分数;之后,采用段共识函数对s段视频片段中的rgb图像的类别分数进行融合产生rgb图像段共识,采用段共识函数对s段视频片段中的分块图像进行融合产生分块图像段共识,采用段共识函数对s段视频片段中的光流场特征的类别分数进行融合产生光流场特征段共识;最后,采用段共识函数对s段视频片段中的rgb图像段共识、分块图像段共识和光流场特征段共识进行加权融合,输出待识别视频的人体行为分类得分。

本发明构建了多流深度学习模型,来验证不同输入模式的作用,以提高人体行为识别的准确性。建立视频帧中局部区域之间的空间融合模型,并将局部融合结果作为输入模式之一。同时,为了捕获视频中的全局表征信息和运动信息,仍然将rgb特征和光流特征作为输入模式之一。本发明首先,将数据集imagenet上的深度网络训练模型作为每个流中基本模型的预训练模型,并将其在数据集ucf-101和数据集hmgb51上做训练,以获得更好的模型;然后再对输出的结果进行融合;分块的网络流与rgb流、光流进行加权融合,得到最终的识别结果。

本发明提出了基于多流深度学习的人体行为识别方法,将人体局部信息与全局信息相结合,通过局部不同特征的精确识别,使人体行为识别更加准确。实验表明,与现有深度学习方法相比,本文提出的方法在数据集ucf101和hmdb51上识别精度分别平均提高了约4.0%和6.2%。

附图说明

图1为本发明基于多流深度学习的人体行为识别方法的流程图;

图2为本发明基于多流深度学习的人体行为识别方法的框架图;

图3本发明中rgb分块模态网络流图;

图4为在数据集ucf-101上不同输入模式的识别结果对比图;

图5为在数据集hmdb51上不同输入模式的识别结果对比图。

具体实施方式

为使本发明实施例的目的、技术方案和优点更加清楚,下面将结合本发明实施例的附图1至5,对本发明实施例的技术方案进行清楚、完整地描述。显然,所描述的实施例是本发明的一部分实施例,而不是全部的实施例。基于所描述的本发明的实施例,本领域普通技术人员所获得的所有其他实施例,都属于本发明保护的范围。

参阅附图1至5,基于多流深度学习的人体行为识别方法,包括以下步骤:

s1、分别建立基于bn-inception网络的全局空域信息网络训练模型、局部空域信息网络训练模型和时域信息网络训练模型;

s2、训练并优化全局空域信息网络训练模型、局部空域信息网络训练模型和时域信息网络训练模型的参数,获取全局空域信息网络模型(图2中lstmcell)、局部空域信息网络模型(图2中空域的convnet)和时域信息网络模型(图2中时域的convnet);

s3、将待识别视频v划分为s段非重叠的视频片段;

s4、在每个视频片段中随机提取一帧rgb图像以及rgb图像的分块图像和光流场特征,rgb图像作为对应视频片段的全局空域信息输入全局空域信息网络模型,分块图像作为对应视频片段的局部空域信息输入局部空域信息网络模型,光流场特征作为对应视频片段的时域信息输入时域信息网络模型;

s5、全局空域信息网络模型输出rgb图像的类别分数,局部空域信息网络模型输出分块图像的类别分数,时域信息网络模型输出光流场特征的类别分数;

s6、采用段共识函数对s段视频片段中的rgb图像的类别分数进行融合产生rgb图像段共识,采用段共识函数对s段视频片段中的分块图像进行融合产生分块图像段共识,采用段共识函数对s段视频片段中的光流场特征的类别分数进行融合产生光流场特征段共识;

s7、采用段共识函数对s段视频片段中的rgb图像段共识、分块图像段共识和光流场特征段共识进行加权融合,输出待识别视频的人体行为分类得分。

如图2所示,本发明的整体大模型包含三个网络结构流,分别为全局空域信息、时域信息和局部空域信息。

自2012年深度学习方法以alexnet为代表,表现出了超凡的性能。而bn-inception网络的出现大大解决了训练收敛问题,用密集成分来近似最优的局部稀疏结构,在很大程度上提升了算法的准确性。本文采用inception网络结构,并在数据集imagenet[文献:dengj,dongw,socherr,etal.imagenet:alarge-scalehierarchicalimagedatabase[c]//computervisionandpatternrecognition,2009.cvpr2009.ieeeconferenceon.ieee,2009:248-255.]上进行预训练。这种网络结构被认为具有更好的准确性和稳定性。

bn-inception网络是googlenet的一种,batchnormalization层的加入,减少了内部神经元数据分布可能发生的变化,而且3×3的卷积层替代较大的5×5卷积层,不仅大大降低了参数的数量,而且提高了运算的速度。表1列出了bn-inception网络的框架结构。

其中,在步骤s3中,将bn-inception在数据集在imagenet上做训练,并将训练后的bn-inception作为预训练模型,并对输入结果进行融合,最终得到一个输出结果。

其中,在步骤s3中,将数据集imagenet上的深度网络训练模型作为每个流中基本模型的预训练模型,并将其在数据集ucf-101和数据集hmgb51上做训练,以获得优化的模型。实验数据集采用当前最流行的两个人体行为识别数据集,ucf101和hmdb51。这两个数据集都是非常具有挑战性的数据集。ucf101数据集包含13320个视频剪辑,分布在101个类中,是最大的动作数据集之一,shipin主要来自youtube网络,主要包括的动作有人和物体交互,人体的肢体动作,人与人之间的动作交互,乐器的操作,以及各种体育运动等。hmpd51数据集共有6766个视频,共51个类,分为3570个训练和1530个测试视频,主要来源于电影、公共数据库以及youtube等网络视频库中。

其中,对于每一个输入片段,在训练阶段,一个输入视频被分为s段,对每一段随机采样得到对应的一个片段;不同片段的类别得分采用段共识函数进行融合来产生段共识;然后对所有模式的预测融合产生最终的预测结果。

其中,对于给定一段视频v,把它按相等间隔分为s段,为{v1,v2,…,vs},视频v的动作识别公式为:c=h(f(g(rand(v1),w),…,g(rand(vs),w)));

其中,f表示段共识函数,c表示类别数,g表示在训练阶段使用的对应类别的真值,l表示损失函数;

最终的损失函数为:其中,y表示训练数据集中的标签,f表示预测的结果,l(y,f)表示训练过程中使用的损失函数;

由h表示预测函数,用于对整个视频的类别进行预测:

其中,所述rgb分块图像的分块方式为:采用由上到下水平平均分配的方法将每一帧rgb图像分为大小相等的三个部分,然后分别将这三个部分的图像输入bn-inception网络框架中得到对应的结果,并将三个结果进行加权融合得到分类得分。

其中,在分块过程中,bn-inception网络框架与rgb网络流共享参数。

其中,在训练阶段,全局空域信息网络流、时域信息网络流和局部空域信息网络流均采用单独训练;在测试阶段,全局空域信息网络流、时域信息网络流和局部空域信息网络流均采用分开测试,最后得出结果再融合。

其中,在训练阶段,rgb图像、rgb分块图像和光流场特征的输入模式的大小都调整为32×32。

其中,在测试阶段,rgb网络流和图像块网络流共享参数。

本发明利用python语言编程实现,使用cuda8的库及opencv库提取光流特征。在训练阶段,所有输入模式的大小(包括rgb图像、光流、rgb块图像)都调整为32×32。初始化学习速率为0.001。在测试阶段,rgb网络流和图像块网络流共享参数。

在网络的输入部分,rgb网络流与光流网络流与tsn方法中相同,但不同的是,本文将分块rgb信息作为一种模态输入网络流中,与其他两种模态进行融合得到识别结果。在分块的过程中,采用了常用的由上到下水平平均分配的方法,对每一帧rgb图像分为大小相等的三个部分,然后分别将三个部分的图像输入本文所提的网络框架中得到对应的结果,并将三个结果进行加权融合。其中,这里采用的网络结构与rgb网络流共享参数,具体的分块方式如图3所示。

利用本发明所提出的方法,首先,将数据集imagenet上的深度网络训练模型作为每个流中基本模型的预训练模型,并将其在数据集ucf-101和数据集hmgb51上做训练,以获得更好的模型。然后再对输出的结果进行融合。分块的网络流与rgb流、光流进行加权融合,得到最终的识别结果。

本发明与现有技术相比较,其实验结果与分析具体如下:

本发明探讨了不同模态数据在行为识别算法中的作用。分别测试ucf101和hmdb51数据集上三种模式及其不同组合的准确率,如图4和图5所示。本发明与当前流行的人体行为识别方法进行了对比,例如基于深度学习的方法,2018年发表在aiai上的atw方法[文献:zangj,wangl,liuz,etal.attention-basedtemporalweightedconvolutionalneuralnetworkforactionrecognition[c]//ifipinternationalconferenceonartificialintelligenceapplicationsandinnovations.springer,cham,2018:97-108.],以及经典的多流方法tsn[文献:wangl,xiongy,wangz,etal.temporalsegmentnetworks:towardsgoodpracticesfordeepactionrecognition[j].2016,22(1):20-36.]和two-stream[文献:朱煜,赵江坤,王逸宁,等.基于深度学习的人体行为识别算法综述.自动化学报,2016,42(6):848-857.]方法,另外,也与传统的非深度学习方法idt[文献:ioffes,szegedyc.batchnormalization:acceleratingdeepnetworktrainingbyreducinginternalcovariateshift[j].2015:448-456.]和bovw[文献:pengx,wangl,wangx,etal.bagofvisualwordsandfusionmethodsforactionrecognition[j].computervision&imageunderstanding,2016,150(c):109-125.]进行了对比。

在ucf101数据集上,如图4,三种模式的准确率都在86%以上:rgb图像-87.6%,rgb块图像-87.6%,光流-86.5%。在tsn方法中[文献:wangl,xiongy,wangz,etal.temporalsegmentnetworks:towardsgoodpracticesfordeepactionrecognition[j].2016,22(1):20-36.]中,同一数据集中warpped光流的准确率仅为86.9%,小于rgb块图像的87.6%。由此,可以推断出局部块空间流与其他模式组合的精确率可以高于tsn中弯曲光流与其他模式组合的精确率。结果表明,当rgb块图像和光流两种模式融合时,性能达到94.4%,三种模式融合的结果为94.8%。不难发现,与单模态或2模态融合相比,三模态数据的融合显著提高了性能。

为了验证该算法的稳定性,本发明在hmdb51数据集上进行了同样的实验,如图5所示。如果输入单模态数据,则准确率低于两种模态融合的结果,并且低于三种模态融合的结果。

最后,通过本发明提出的算法,融合三种不同的模式生成人类活动识别算法的最终结果。为了更客观地评价该算法的性能,我们在不同的数据集上与现有的主流算法进行了比较。如表2所示,总结了该算法和对比方法在数据集hmdb51和数据集ucf101上的性能。

对比法主要包括传统方法和深度学习方法。传统方法主要包括改进的轨迹模型[文献:wangh,schmidc.actionrecognitionwithimprovedtrajectories[c]//ieeeinternationalconferenceoncomputervision.ieee,2014:3551-3558.]、视觉词袋和融合方法[文献:pengx,wangl,wangx,etal.bagofvisualwordsandfusionmethodsforactionrecognition[j].computervision&imageunderstanding,2016,150(c):109-125.];深层学习方法主要包括两个流[文献:朱煜,赵江坤,王逸宁,等.基于深度学习的人体行为识别算法综述.自动化学报,2016,42(6):848-857.]、tsn(rgb+flow)[文献:[12]wangl,xiongy,wangz,etal.temporalsegmentnetworks:towardsgoodpracticesfordeepactionrecognition[j].2016,22(1):20-36.]、tsn(rgb+flow+war.flow)[文献:wangl,xiongy,wangz,etal.temporalsegmentnetworks:towardsgoodpracticesfordeepactionrecognition[j].2016,22(1):20-36.]和基于注意力的时间加权的卷积神经网路(atw)[文献:zangj,wangl,liuz,etal.attention-basedtemporalweightedconvolutionalneuralnetworkforactionrecognition[c]//ifipinternationalconferenceonartificialintelligenceapplicationsandinnovations.springer,cham,2018:97-108.]。本发明的算法优于hmgb51数据集和ucf101数据集上的其他方法。该方法在hmgb51和ucf101上的准确率分别为70.6%和94.8%。这表明本地块流的局部特征的有效性。特别与tsn(rgb+flow+war.flow)相比,发现局部块流能够代替弯曲光流。

从对比结果可以看到,由于不同数据集存在一定的差异性,造成同一种方法在不同数据集上得到的结果不同,甚至出现较大的差别。例如数据集ucf101和hmdb51两个数据集,不仅是我们的方法在这两个数据集上的实验结果具有一定的差别,在ucf101上的准确率为94.8%,而在hmdb51上的准确率却只有70.6%,其他对方方法在这两个数据集上也存在这样的差别。这主要是由于数据集本身的特性,以及方法本身的泛化能力不足造成的。

表2在数据集ucf-101和hmdb-5上的比较

本发明提供基于多流深度学习的人体行为识别方法,该方法结合了全局空域信息、时域信息和局部空域信息。一方面,融合了局部时空信息。将rgb图像分成三个部分,用于提取视频帧局部的细节。另一方面,实现了多种模态融合。通过考虑不同的模态信息,将三种模式的结果融合,得到识别结果。为了验证不同模态信息在行为识别中的作用,本发明比较了不同模态信息与融合后识别结果的正确率,并与现有的主流算法进行了比较,证明了本发明所提算法的有效性。

最后说明的是,以上实施例仅用以说明本发明的技术方案而非限制,本领域普通技术人员对本发明的技术方案所做的其他修改或者等同替换,只要不脱离本发明技术方案的精神和范围,均应涵盖在本发明的权利要求范围当中。

当前第1页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1